diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-26 06:00:17 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-26 06:04:42 -0700 |
| commit | 544436bbdf279dd4be68ad71536ea0488258aa07 (patch) | |
| tree | ff85ac9a4f472be34b739981c7388e218a4e46ec /tensorflow/contrib/slim | |
| parent | ed37c8a8a07734a4eb13e14d7d7b67c81a2968b7 (diff) | |
Automated g4 rollback of changelist 202000826
PiperOrigin-RevId: 202115471
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))) |