diff options
Diffstat (limited to 'tensorflow/contrib/training/python/training/training_test.py')
-rw-r--r-- | tensorflow/contrib/training/python/training/training_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/training/python/training/training_test.py b/tensorflow/contrib/training/python/training/training_test.py index 918c1da018..c0e79aa798 100644 --- a/tensorflow/contrib/training/python/training/training_test.py +++ b/tensorflow/contrib/training/python/training/training_test.py @@ -310,7 +310,7 @@ class TrainTest(tf.test.TestCase): tf.set_random_seed(2) train_op = self.create_train_op() - model_variables = tf.all_variables() + model_variables = tf.global_variables() model_path = os.path.join(logdir1, 'model.ckpt-300') assign_fn = tf.contrib.framework.assign_from_checkpoint_fn( |