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

checkpointing fix

Browse Source
This commit is contained in:
Shadi Endrawis
2018-10-07 20:06:08 +03:00
parent 5c4f9d58dd
commit 364168490f
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

@@ -220,7 +220,7 @@ class GraphManager(object):
self.set_session(self.sess) self.set_session(self.sess)
else: else:
self.variables_to_restore = tf.global_variables() 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) self.checkpoint_saver = tf.train.Saver(self.variables_to_restore)
# regular session # regular session