From 433bc3e27b53e582c92400f0518909851ed667a0 Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Thu, 13 Sep 2018 19:35:28 +0000 Subject: [PATCH] standardizing variable access --- 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 ddbafd4..d13259b 100644 --- a/rl_coach/graph_managers/graph_manager.py +++ b/rl_coach/graph_managers/graph_manager.py @@ -479,7 +479,7 @@ class GraphManager(object): checkpoint = tf.train.get_checkpoint_state(checkpoint_dir) screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path)) variables = {} - for var_name, _ in tf.contrib.framework.list_variables(self.task_parameters.checkpoint_restore_dir): + for var_name, _ in tf.contrib.framework.list_variables(checkpoint_dir): # Load the variable var = tf.contrib.framework.load_variable(checkpoint_dir, var_name)