浏览代码

Fix bug when academy max steps is set to nonzero value. (#1195)

/develop-generalizationTraining-TrainerController
GitHub 6 年前
当前提交
af7de3ca
共有 1 个文件被更改,包括 5 次插入2 次删除
  1. 7
      ml-agents/mlagents/trainers/trainer_controller.py

7
ml-agents/mlagents/trainers/trainer_controller.py


# If any lessons were incremented or the environment is
# ready to be reset
if (self.meta_curriculum
and any(lessons_incremented.values())
or self.env.global_done):
and any(lessons_incremented.values())):
curr_info = self._reset_env()
for brain_name, trainer in self.trainers.items():
trainer.end_episode()

elif self.env.global_done:
curr_info = self._reset_env()
for brain_name, trainer in self.trainers.items():
trainer.end_episode()
# Decide and take an action
take_action_vector, \

正在加载...
取消
保存