Skip to content

GitLab

Commit 8c95d1b9 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #1558 from mbz/master 

Fixing the initialization/loading bug.
parents 35f6f9c9 34af79db
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
video_prediction/prediction_train.py
View file @ 8c95d1b9
...@@ -204,6 +204,8 @@ def main(unused_argv): ...@@ -204,6 +204,8 @@ def main(unused_argv):
# Make training session. # Make training session.
sess = tf.InteractiveSession() sess = tf.InteractiveSession()
sess.run(tf.global_variables_initializer())
summary_writer = tf.summary.FileWriter( summary_writer = tf.summary.FileWriter(
FLAGS.event_log_dir, graph=sess.graph, flush_secs=10) FLAGS.event_log_dir, graph=sess.graph, flush_secs=10)
...@@ -211,7 +213,6 @@ def main(unused_argv): ...@@ -211,7 +213,6 @@ def main(unused_argv):
saver.restore(sess, FLAGS.pretrained_model) saver.restore(sess, FLAGS.pretrained_model)
tf.train.start_queue_runners(sess) tf.train.start_queue_runners(sess)
sess.run(tf.global_variables_initializer())
tf.logging.info('iteration number, cost') tf.logging.info('iteration number, cost')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment