diff options
Diffstat (limited to 'tensorflow/python/training/monitored_session_test.py')
-rw-r--r-- | tensorflow/python/training/monitored_session_test.py | 36 |
1 files changed, 0 insertions, 36 deletions
diff --git a/tensorflow/python/training/monitored_session_test.py b/tensorflow/python/training/monitored_session_test.py index 3806056f01..159b2d5c16 100644 --- a/tensorflow/python/training/monitored_session_test.py +++ b/tensorflow/python/training/monitored_session_test.py @@ -282,42 +282,6 @@ class MonitoredTrainingSessionTest(test.TestCase): is_chief=True, checkpoint_dir=logdir) as session: self.assertEqual(2, session.run(gstep)) - def test_save_checkpoint_steps(self): - logdir = _test_dir(self.get_temp_dir(), 'test_save_checkpoint_steps') - with ops.Graph().as_default(): - gstep = variables_lib.get_or_create_global_step() - new_gstep = state_ops.assign_add(gstep, 1) - with monitored_session.MonitoredTrainingSession( - is_chief=True, - checkpoint_dir=logdir, - save_checkpoint_steps=100, - log_step_count_steps=10) as session: - for _ in range(100): - session.run(new_gstep) - # A restart will find the checkpoint and recover automatically. - with monitored_session.MonitoredTrainingSession( - is_chief=True, checkpoint_dir=logdir) as session: - self.assertEqual(100, session.run(gstep)) - - def test_save_checkpoint_secs(self): - logdir = _test_dir(self.get_temp_dir(), 'test_save_checkpoint_secs') - with ops.Graph().as_default(): - gstep = variables_lib.get_or_create_global_step() - new_gstep = state_ops.assign_add(gstep, 1) - with monitored_session.MonitoredTrainingSession( - is_chief=True, - checkpoint_dir=logdir, - save_checkpoint_secs=0.1, - log_step_count_steps=10) as session: - session.run(new_gstep) - time.sleep(0.2) - for _ in range(10): - session.run(new_gstep) - # A restart will find the checkpoint and recover automatically. - with monitored_session.MonitoredTrainingSession( - is_chief=True, checkpoint_dir=logdir) as session: - self.assertEqual(11, session.run(gstep)) - def test_summaries_steps(self): logdir = _test_dir(self.get_temp_dir(), 'test_summaries_steps') with ops.Graph().as_default(): |