gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 19:50:17 +01:00
Code Issues Projects Releases Wiki Activity

standardizing variable access

Browse Source
This commit is contained in:
Zach Dwiel
2018-09-13 19:35:28 +00:00
committed by zach dwiel
parent e34b9ae9cf
commit 433bc3e27b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/graph_managers/graph_manager.py
View File

@@ -479,7 +479,7 @@ class GraphManager(object):
checkpoint = tf.train.get_checkpoint_state(checkpoint_dir) checkpoint = tf.train.get_checkpoint_state(checkpoint_dir)
screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path)) screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path))
variables = {} 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 # Load the variable
var = tf.contrib.framework.load_variable(checkpoint_dir, var_name) var = tf.contrib.framework.load_variable(checkpoint_dir, var_name)