train.impala

Full path: schola.scripts.rllib.train.impala

schola.scripts.rllib.train.impala

impala

impala(args=RLlibScriptArgs(training_settings=TrainingSettings(timesteps=3000, learning_rate=0.0003, minibatch_size=128, train_batch_size_per_learner=256, num_sgd_iter=5, gamma=0.99), algorithm_settings=PPOSettings(gae_lambda=0.95, clip_param=0.2, use_gae=True), logging_settings=LoggingSettings(schola_verbosity=0, rllib_verbosity=1), resume_settings=ResumeSettings(resume_from=None), network_architecture_settings=NetworkArchitectureSettings(fcnet_hiddens=[512, 512], activation=&lt;ActivationFunctionEnum.ReLU: 'relu'&gt;, use_attention=False, attention_dim=64), resource_settings=ResourceSettings(num_gpus=0, num_cpus=1, num_learners=0, num_cpus_for_main_process=1, num_cpus_per_learner=1, num_gpus_per_learner=0, using_cluster=False), checkpoint_settings=CheckpointArgs(enable_checkpoints=False, checkpoint_dir=WindowsPath('ckpt'), save_freq=100000, name_prefix_override=None, export_onnx=False, save_final_policy=False), environment_settings=EnvironmentArgs(simulator=UnrealEditorSimulatorArgs(), protocol=gRPCProtocolArgs(port=None, url='localhost', environment_start_timeout=45))), simulator_type='editor', executable=UnrealExecutableCommandLinePlaceholder(executable_path=None, disable_script=True, headless=False, map=None, fps=None, display_logs=True), impala=IMPALASettings(vtrace=True, vtrace_clip_rho_threshold=1.0, vtrace_clip_pg_rho_threshold=1.0), editor=UnrealEditorSimulatorArgs())

Parameters

args (Annotated[RLlibScriptArgs, Parameter(name=('*',))])

simulator_type (Annotated[Literal['editor', 'executable'], Parameter(group=('Simulator (Choose One)',))])

executable (Annotated[UnrealExecutableSimulatorArgs, Parameter(name=('*',), group=('Executable Parameters',))])

impala (Annotated[IMPALASettings, Parameter(name=('*',), group=('IMPALA Parameters',))])

editor (Annotated[UnrealEditorSimulatorArgs, Parameter(show=False)])