60b1f9eb915691fbac1e5d1edbcf295af628e131,tests/discriminators/pyramid_discriminator_test.py,PyramidDiscriminatorTest,test_create,#PyramidDiscriminatorTest#,28

Before Change

            remove_d_config["discriminator"] = None
            remove_d_config["loss"] = None
            remove_d_config["trainer"] = None
            gan = hg.GAN(config = remove_d_config, graph = {
                "x": tf.constant(1, shape=[1,4,4,1], dtype=tf.float32)
            })
            discriminator = PyramidDiscriminator(gan, config)
            gan.create()
            net = discriminator.create()
            self.assertEqual(int(net.get_shape()[1]), 7)

After Change

            remove_d_config["discriminator"] = None
            remove_d_config["loss"] = None
            remove_d_config["trainer"] = None
            gan = hg.GAN(config = remove_d_config, inputs = MockInput())
            discriminator = PyramidDiscriminator(gan, config)
            gan.create()
            net = discriminator.create()
            self.assertEqual(int(net.get_shape()[1]), 112)

Instances