Merge pull request tensorflow#1558 from mbz/master

Fixing the initialization/loading bug.
envin3 · Jun 15, 2017 · 8c95d1b · 8c95d1b
2 parents 35f6f9c + 34af79d
commit 8c95d1b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/video_prediction/prediction_train.py b/video_prediction/prediction_train.py
@@ -204,14 +204,15 @@ def main(unused_argv):
 
   # Make training session.
   sess = tf.InteractiveSession()
+  sess.run(tf.global_variables_initializer())
+
   summary_writer = tf.summary.FileWriter(
       FLAGS.event_log_dir, graph=sess.graph, flush_secs=10)
 
   if FLAGS.pretrained_model:
     saver.restore(sess, FLAGS.pretrained_model)
 
   tf.train.start_queue_runners(sess)
-  sess.run(tf.global_variables_initializer())
 
   tf.logging.info('iteration number, cost')