Fixing the initialization/loading bug.

The code currently loads the checkpoint and then initializes the variables resulting to random weights. 
Swapping the order fixes the loading checkpoint issue.

Author: mbz

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')