浏览代码

Fix crash in construct_curr_info when next_info doesn't have any agents (#2549)

Fixes #1687
/develop-gpu-test
GitHub 5 年前
当前提交
b7e12a37
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 3
      ml-agents/mlagents/trainers/rl_trainer.py

3
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:
# Check if memories exists (i.e. next_info is not empty) before attempting vstack
if self.policy.use_recurrent and memories:
memories = np.vstack(memories)
curr_info = BrainInfo(
visual_observations,

正在加载...
取消
保存