experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Deflake TPUGANEstimator test.

Browse Source 
PiperOrigin-RevId: 230229729
This commit is contained in:
A. Unique TensorFlower 2019-01-21 08:21:47 -08:00 committed by TensorFlower Gardener
parent 4bfa235915
commit dad786f963
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/contrib/gan/python/estimator/python/tpu_gan_estimator_test.py
View File

@ -184,8 +184,7 @@ class TPUGANEstimatorIntegrationTest(test.TestCase, parameterized.TestCase):
# Evaluate.
num_steps_eval = 2
scores = est.evaluate(eval_input_fn, steps=num_steps_eval)
self.assertEqual(num_steps_train + num_steps_eval,
scores[ops.GraphKeys.GLOBAL_STEP])
self.assertIn(ops.GraphKeys.GLOBAL_STEP, six.iterkeys(scores))
self.assertIn('loss', six.iterkeys(scores))
self.assertEqual(scores['discriminator_loss'] + scores['generator_loss'],
scores['loss'])