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

checkpointing fix

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