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

[tfgan] Fix assertion in regularization unittest

Browse Source 
`self.assertTrue(3.0)` always evaluates to `True`.
This PR fixes assertion so it will correctly check the loss value.
This commit is contained in:
Lukas Geiger 2018-07-15 23:36:37 +02:00
parent e5945c0014
commit 006b8faeb7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/contrib/gan/python/train_test.py
View File

@ -447,8 +447,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)