45db0508379d807a472df7ef4fb007876f52a6f4,examples/2d-test.py,,custom_generator,#Any#Any#Any#,65

Before Change

def custom_generator(config, gan, net):
    net = linear(net, 128, scope="g_lin_proj")
    net = batch_norm_1(gan.config.batch_size, name="g_bn_1")(net)
    net = tf.nn.relu(net)
    net = linear(net, 2, scope="g_lin_proj3")
    net = tf.tanh(net)
    return [net]

After Change

def custom_generator(config, gan, net):
    net = linear(net, 128, scope="g_lin_proj")
    for i in range(24):
        net = tf.nn.crelu(net)
        net = linear(net, 128, scope="g_lin"+str(i))
    net = tf.nn.crelu(net)
    net = linear(net, 2, scope="g_lin_proj3")
    net = tf.tanh(net)
    return [net]

Instances