Bug fix: Session.initialize_variables now always uses the correct graph
@@ -24,6 +24,7 @@ class Session(object):
self.saver = None
def initialize_variables(self):
+ with self.graph.as_default():
self._session.run(tf.global_variables_initializer())
self.saver = tf.train.Saver(max_to_keep=100, allow_empty=True)
