浏览代码

Fix crash when next_info is empty and using recurrent

/develop-gpu-test
Ervin Teng 5 年前
当前提交
4cb340b5
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/rl_trainer.py

2
ml-agents/mlagents/trainers/rl_trainer.py


agent_brain_info.previous_text_actions[agent_index]
)
action_masks.append(agent_brain_info.action_masks[agent_index])
if self.policy.use_recurrent:
if self.policy.use_recurrent and memories:
memories = np.vstack(memories)
curr_info = BrainInfo(
visual_observations,

正在加载...
取消
保存