Merge pull request #1558 from mbz/master

Fixing the initialization/loading bug.

Merge pull request #1558 from mbz/master
Fixing the initialization/loading bug.
8c95d1b9 · Neal Wu · GitHub · 35f6f9c9 · 34af79db · 8c95d1b9
Commit 8c95d1b9 authored Jun 15, 2017 by Neal Wu Committed by GitHub Jun 15, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

video_prediction/prediction_train.py video_prediction/prediction_train.py +2 -1

No files found.
--- a/video_prediction/prediction_train.py
+++ b/video_prediction/prediction_train.py
@@ -204,6 +204,8 @@ 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)

@@ -211,7 +213,6 @@ def main(unused_argv):
    saver.restore(sess, FLAGS.pretrained_model)

  tf.train.start_queue_runners(sess)
-  sess.run(tf.global_variables_initializer())

  tf.logging.info('iteration number, cost')