File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
ml-agents/mlagents/trainers Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff 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 , \
You can’t perform that action at this time.
0 commit comments