diff options
author | 2018-06-14 02:14:42 -0700 | |
---|---|---|
committer | 2018-06-14 02:17:24 -0700 | |
commit | 8d9787bed57f1dd5d697ff847cd5598ecc032620 (patch) | |
tree | 26accb03641e90b4236edfc1da587cd8f4638666 /tensorflow/contrib/slim | |
parent | 0b8c5806f4f1d3a47b30bf203b3e456f036b0adc (diff) |
Automated g4 rollback of changelist 200467580
PiperOrigin-RevId: 200525639
Diffstat (limited to 'tensorflow/contrib/slim')
-rw-r--r-- | tensorflow/contrib/slim/python/slim/learning_test.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/slim/python/slim/learning_test.py b/tensorflow/contrib/slim/python/slim/learning_test.py index 6bd55e7a24..831c6e427a 100644 --- a/tensorflow/contrib/slim/python/slim/learning_test.py +++ b/tensorflow/contrib/slim/python/slim/learning_test.py @@ -520,6 +520,8 @@ class TrainTest(test.TestCase): run_root = glob.glob(os.path.join(dump_root, 'run_*'))[-1] dump = debug_data.DebugDumpDir(run_root) + self.assertAllEqual(0, + dump.get_tensors('global_step', 0, 'DebugIdentity')[0]) def testTrainWithTrace(self): logdir = os.path.join( @@ -545,7 +547,7 @@ class TrainTest(test.TestCase): log_every_n_steps=10, trace_every_n_steps=100) self.assertIsNotNone(loss) - for trace_step in [0, 100, 200]: + for trace_step in [1, 101, 201]: trace_filename = 'tf_trace-%d.json' % trace_step self.assertTrue(os.path.isfile(os.path.join(logdir, trace_filename))) |