diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-18 01:43:45 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-18 01:43:52 -0700 |
| commit | 3a576d3a2847cce68c4c4565f8a1124d7421ca3e (patch) | |
| tree | 0aa639263cc2cd2b5fa0704ae43e445066e6b934 /tensorflow/contrib/gan | |
| parent | 5d6aec5318fba138bdf47ff0fbee035dc4d6e04a (diff) | |
| parent | 006b8faeb79c8b9329bd600390dbda888e9df226 (diff) | |
Merge pull request #20820 from lgeiger:tfgan-fix-test
PiperOrigin-RevId: 205042372
Diffstat (limited to 'tensorflow/contrib/gan')
| -rw-r--r-- | tensorflow/contrib/gan/python/train_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/gan/python/train_test.py b/tensorflow/contrib/gan/python/train_test.py index 06681eaf83..93a12af944 100644 --- a/tensorflow/contrib/gan/python/train_test.py +++ b/tensorflow/contrib/gan/python/train_test.py @@ -613,8 +613,8 @@ class GANLossTest(test.TestCase): reg_loss_gen_np = reg_loss.generator_loss.eval() reg_loss_dis_np = reg_loss.discriminator_loss.eval() - self.assertTrue(3.0, reg_loss_gen_np - no_reg_loss_gen_np) - self.assertTrue(3.0, reg_loss_dis_np - no_reg_loss_dis_np) + self.assertEqual(3.0, reg_loss_gen_np - no_reg_loss_gen_np) + self.assertEqual(2.0, reg_loss_dis_np - no_reg_loss_dis_np) def test_regularization_gan(self): self._test_regularization_helper(get_gan_model) |