浏览代码

added ppo/sac_policy attributes to keep up with master

/develop/magic-string
Andrew Cohen 5 年前
当前提交
6a4e7cf9
共有 2 个文件被更改,包括 5 次插入0 次删除
  1. 2
      ml-agents/mlagents/trainers/ppo/trainer.py
  2. 3
      ml-agents/mlagents/trainers/sac/trainer.py

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


for _reward_signal in policy.reward_signals.keys():
self.collected_rewards[_reward_signal] = {}
self.ppo_policy = policy
return policy

3
ml-agents/mlagents/trainers/sac/trainer.py


self.update_buffer.num_experiences
)
)
self.sac_policy = policy
return policy
def update_sac_policy(self) -> None:

正在加载...
取消
保存