浏览代码

forgot one

/develop/remove-unecessary-lr
vincentpierre 3 年前
当前提交
5d384292
共有 1 个文件被更改,包括 1 次插入3 次删除
  1. 4
      ml-agents/mlagents/trainers/ppo/optimizer_torch.py

4
ml-agents/mlagents/trainers/ppo/optimizer_torch.py


self.trainer_settings.max_steps,
)
self.optimizer = torch.optim.Adam(
params, lr=self.trainer_settings.hyperparameters.learning_rate
)
self.optimizer = torch.optim.Adam(params)
self.stats_name_to_update_name = {
"Losses/Value Loss": "value_loss",
"Losses/Policy Loss": "policy_loss",

正在加载...
取消
保存