diff options
Diffstat (limited to 'tensorflow/contrib/training/python/training/evaluation_test.py')
| -rw-r--r-- | tensorflow/contrib/training/python/training/evaluation_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/training/python/training/evaluation_test.py b/tensorflow/contrib/training/python/training/evaluation_test.py index 927f6ab75a..3d83aec94e 100644 --- a/tensorflow/contrib/training/python/training/evaluation_test.py +++ b/tensorflow/contrib/training/python/training/evaluation_test.py @@ -51,7 +51,7 @@ class CheckpointIteratorTest(tf.test.TestCase): saver = tf.train.Saver() # Saves the global step. with self.test_session() as session: - session.run(tf.initialize_all_variables()) + session.run(tf.global_variables_initializer()) save_path = os.path.join(checkpoint_dir, 'model.ckpt') saver.save(session, save_path, global_step=global_step) @@ -81,7 +81,7 @@ class CheckpointIteratorTest(tf.test.TestCase): target='', config=tf.ConfigProto(device_count={'CPU': 2})) as session: - session.run(tf.initialize_all_variables()) + session.run(tf.global_variables_initializer()) save_path = os.path.join(checkpoint_dir, 'model.ckpt') saver.save(session, save_path, global_step=global_step) |