浏览代码

Add GAIL learning rate to TB

/develop/decaygail
Ervin Teng 4 年前
当前提交
7b41e5d6
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/torch/components/reward_providers/gail_reward_provider.py

1
ml-agents/mlagents/trainers/torch/components/reward_providers/gail_reward_provider.py


self.optimizer.zero_grad()
loss.backward()
self.optimizer.step()
stats_dict["Policy/GAIL Learning Rate"] = decay_lr
return stats_dict
def get_modules(self):

正在加载...
取消
保存