浏览代码

lstm runs with coma

/develop/action-slice
Andrew Cohen 4 年前
当前提交
4c56e6ad
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 4
      ml-agents/mlagents/trainers/coma/optimizer_torch.py

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


all_value_tensors,
all_baseline_tensors,
all_next_value_mem,
next_value_mem,
next_value_mem,
next_baseline_mem,
)

)
next_value_estimates, _ = self.critic.critic_pass(
all_next_obs, next_value_mem, sequence_length=batch.num_experiences
all_next_obs, next_value_mem, sequence_length=1
)
for name, estimate in baseline_estimates.items():

正在加载...
取消
保存