Function run_trainer

tests/trainer/trainer_test.py:563–578

(config: Config, stop_event=None)

def run_trainer(config: Config, stop_event=None) -> None:
    ray.init(
        namespace=config.ray_namespace,
        runtime_env={
            "env_vars": {
                LOG_DIR_ENV_VAR: config.log.save_dir,
                LOG_LEVEL_ENV_VAR: "INFO",
            }
        },
    )
    try:
        train(config)
    finally:
        if stop_event:
            stop_event.set()
        ray.shutdown()


def run_explorer(config: Config, stop_event=None) -> None: