Merge pull request #20820 from lgeiger:tfgan-fix-test

PiperOrigin-RevId: 205042372
This commit is contained in:
TensorFlower Gardener 2018-07-18 01:43:45 -07:00
commit 3a576d3a28

View File

@ -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)