cifar10/config.json

{
    "img_size": 32,
    "batch_size": 128,
    "g_lr": 0.0002,
    "g_beta1": 0.5,
    "g_beta2": 0.99,
    "noise_dim": 128,
    "g_initializer": "orthogonal",
    "g_dim": [
        1024,
        256,
        64
    ],
    "g_heads": [
        4,
        4,
        4
    ],
    "g_mlp": [
        512,
        512,
        512
    ],
    "d_initializer": "orthogonal",
    "d_enc_dim": [
        128,
        256
    ],
    "d_out_dim": [
        512,
        1024
    ],
    "d_heads": 4,
    "d_mlp": 512,
    "d_lr": 0.0002,
    "d_beta1": 0.5,
    "d_beta2": 0.99,
    "gp_weight": 0.0001,
    "policy": "color,translation",
    "fid_batch_size": 50,
    "gen_batch_size": 50,
    "loss": "nsl",
    "ema_decay": 0.999,
    "bcr": true,
    "cr_weight": 1.0,
    "dec_dim": false,
    "n_fid_real": 2500,
    "n_fid_gen": 2500,
    "plot_size": 5.2,
    "test_seed": 77
}