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)