浏览代码

Fix memory leak

/develop-newnormalization
Ervin Teng 5 年前
当前提交
c2b729a6
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/agent_processor.py

1
ml-agents/mlagents/trainers/agent_processor.py


agent_id, training_length=self.policy.sequence_length
)
self.trainer.process_trajectory(trajectory)
self.processing_buffer[agent_id].reset_agent()
elif not next_info.local_done[next_idx]:
if agent_id not in self.episode_steps:
self.episode_steps[agent_id] = 0

正在加载...
取消
保存