浏览代码

Zero out memory

/develop/0memories
Ervin Teng 5 年前
当前提交
649c4185
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/ppo/policy.py

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


feed_dict[model.visual_in[i]] = mini_batch["visual_obs%d" % i]
if self.use_recurrent:
mem_in = [
mini_batch["memory"][i]
np.zeros((self.m_size))
for i in range(0, len(mini_batch["memory"]), self.sequence_length)
]
feed_dict[model.memory_in] = mem_in
正在加载...
取消
保存