Skip to content

Commit 0f0379e

Browse files
author
Deric Pang
authored
Fix bug when academy max steps is set to nonzero value. (#1195)
1 parent b0f3bfe commit 0f0379e

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

ml-agents/mlagents/trainers/trainer_controller.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -336,14 +336,17 @@ def start_learning(self):
336336
# If any lessons were incremented or the environment is
337337
# ready to be reset
338338
if (self.meta_curriculum
339-
and any(lessons_incremented.values())
340-
or self.env.global_done):
339+
and any(lessons_incremented.values())):
341340
curr_info = self._reset_env()
342341
for brain_name, trainer in self.trainers.items():
343342
trainer.end_episode()
344343
for brain_name, changed in lessons_incremented.items():
345344
if changed:
346345
self.trainers[brain_name].reward_buffer.clear()
346+
elif self.env.global_done:
347+
curr_info = self._reset_env()
348+
for brain_name, trainer in self.trainers.items():
349+
trainer.end_episode()
347350

348351
# Decide and take an action
349352
take_action_vector, \

0 commit comments

Comments
 (0)