diff --git a/rl_coach/graph_managers/graph_manager.py b/rl_coach/graph_managers/graph_manager.py index 4756b80..03b39f6 100644 --- a/rl_coach/graph_managers/graph_manager.py +++ b/rl_coach/graph_managers/graph_manager.py @@ -386,7 +386,7 @@ class GraphManager(object): if result.game_over: self.handle_episode_ended() - # TODO: why not just reset right now? + self.reset_required = True if keep_networks_in_sync: self.sync_graph()