From d32d9092384415bdd6347e388c80b05f8fb8a83c Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Thu, 4 Oct 2018 11:52:24 -0400 Subject: [PATCH] move only invocation of GraphManager.handle_episode_ended inline --- rl_coach/graph_managers/graph_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()