gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-09 15:22:36 +02:00
Code Issues Projects Releases Wiki Activity

standardizing variable access

Browse Source
This commit is contained in:
Zach Dwiel authored and zach dwiel committed 2018-10-23 16:40:33 -04:00
1 parent e34b9ae9cf
commit 433bc3e27b
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
rl_coach/graph_managers/graph_manager.py
+1 -1
View File
@@ -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)