Merge https://github.com/tensorflow/models

fe794eab · Ivan Bogatyy · 0965f53d · 51fcc99b · fe794eab
Commit fe794eab authored Mar 16, 2017 by Ivan Bogatyy
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/video_prediction/prediction_train.py
+++ b/video_prediction/prediction_train.py
@@ -196,7 +196,7 @@ def main(unused_argv):
  print 'Constructing saver.'
  # Make saver.
  saver = tf.train.Saver(
-      tf.get_collection(tf.GraphKeys.VARIABLES), max_to_keep=0)
+      tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES), max_to_keep=0)

  # Make training session.
  sess = tf.InteractiveSession()