From 364168490fb0ae4554885bd441a0d9c44baf3dea Mon Sep 17 00:00:00 2001 From: Shadi Endrawis Date: Sun, 7 Oct 2018 20:06:08 +0300 Subject: [PATCH] checkpointing fix --- 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 aa84dd3..53525ea 100644 --- a/rl_coach/graph_managers/graph_manager.py +++ b/rl_coach/graph_managers/graph_manager.py @@ -220,7 +220,7 @@ class GraphManager(object): self.set_session(self.sess) else: self.variables_to_restore = tf.global_variables() - self.variables_to_restore = [v for v in self.variables_to_restore if '/online' in v.name] + # self.variables_to_restore = [v for v in self.variables_to_restore if '/online' in v.name] TODO: is this necessary? self.checkpoint_saver = tf.train.Saver(self.variables_to_restore) # regular session