89fce6b0e5baf9c3b844c068d37956912f9f47e4,hypergan/discriminators/fully_connected_discriminator.py,FullyConnectedDiscriminator,build,#FullyConnectedDiscriminator#Any#,12

Before Change

        print("[fully connected discriminator] creating FC layer from ", net)
        net = ops.linear(net, 512)//config.features or ops.shape(net)[-1])
        net = activation(net)
        net = ops.linear(net, 512)//config.features or ops.shape(net)[-1])
        //net = ops.linear(net, config.features or ops.shape(net)[-1])
        if final_activation:
            net = final_activation(net)

After Change

        print("[fully connected discriminator] creating FC layer from ", net)
        net = ops.linear(net, config.features or ops.shape(net)[-1])
        for i in range(config.layers or 1):
            net = self.layer_regularizer(net)
            net = activation(net)
            net = ops.linear(net, config.features or ops.shape(net)[-1])

        if final_activation:
            net = self.layer_regularizer(net)
            net = final_activation(net)

        self.sample = net

Instances