浏览代码

[cleanup] Remove unused param keys (#4067)

/MLA-1734-demo-provider
GitHub 5 年前
当前提交
7229214c
共有 2 个文件被更改,包括 0 次插入21 次删除
  1. 20
      ml-agents/mlagents/trainers/ppo/trainer.py
  2. 1
      ml-agents/mlagents/trainers/trainer/trainer.py

20
ml-agents/mlagents/trainers/ppo/trainer.py


super(PPOTrainer, self).__init__(
brain_name, trainer_settings, training, run_id, reward_buff_cap
)
self.param_keys = [
"batch_size",
"beta",
"buffer_size",
"epsilon",
"hidden_units",
"lambd",
"learning_rate",
"max_steps",
"normalize",
"num_epoch",
"num_layers",
"time_horizon",
"sequence_length",
"summary_freq",
"use_recurrent",
"memory_size",
"output_path",
"reward_signals",
]
self.hyperparameters: PPOSettings = cast(
PPOSettings, self.trainer_settings.hyperparameters
)

1
ml-agents/mlagents/trainers/trainer/trainer.py


:str run_id: The identifier of the current run
:int reward_buff_cap:
"""
self.param_keys: List[str] = []
self.brain_name = brain_name
self.run_id = run_id
self.trainer_settings = trainer_settings

正在加载...
取消
保存