ACRunner

class maze.train.trainers.common.actor_critic.actor_critic_runners.ACRunner(state_dict_dump_file: str, dump_interval: Optional[int], spaces_config_dump_file: str, normalization_samples: int, trainer_class: Union[str, type], concurrency: int, eval_concurrency: int)

Abstract baseclass of AC runners.

concurrency: int

Number of concurrently executed environments.

abstract create_distributed_env(env_factory: Callable[], Union[maze.core.env.maze_env.MazeEnv, maze.core.env.structured_env.StructuredEnv]], concurrency: int, logging_prefix: str) → maze.train.parallelization.vector_env.structured_vector_env.StructuredVectorEnv

The dev and local runner implement the setup of the distribution env

eval_concurrency: int

Number of concurrent evaluation envs

setup(cfg: omegaconf.DictConfig) → None

(overrides TrainingRunner)

See setup().

trainer_class: Union[str, type]

The actor critic trainer class to be used.