def test_create(self): with self.test_session(): gan = mock_gan() loss = CramerLoss(gan, loss_config) d_loss, g_loss = loss.create() d_shape = gan.ops.shape(d_loss) g_shape = gan.ops.shape(g_loss) self.assertEqual(sum(d_shape), 0) self.assertEqual(sum(g_shape), 0)
def test_config(self): with self.test_session(): loss = CramerLoss(hg.GAN(), loss_config) self.assertTrue(loss.config.test)
def test_config(self): with self.test_session(): gan = mock_gan() loss = CramerLoss(gan, loss_config) self.assertTrue(loss.config.test)
def test_config(self): with self.test_session(): gan = mock_gan(config=remove_d_config, inputs=MockInput()) loss = CramerLoss(gan, loss_config) self.assertTrue(loss.config.test)