(config)
| 91 | |
| 92 | @hydra.main(config_path='config', config_name='ppo_trainer_split', version_base=None) |
| 93 | def main(config): |
| 94 | if not ray.is_initialized(): |
| 95 | # this is for local ray cluster |
| 96 | ray.init(runtime_env={'env_vars': {'TOKENIZERS_PARALLELISM': 'true', 'NCCL_DEBUG': 'WARN'}}) |
| 97 | |
| 98 | ray.get(main_task.remote(config)) |
| 99 | |
| 100 | |
| 101 | @ray.remote |
no test coverage detected