{ "img_size": 128, "batch_size": 64, "g_lr": 0.0002, "g_beta1": 0.5, "g_beta2": 0.99, "noise_dim": 128, "g_initializer": "orthogonal", "g_dim": [ 1024, 512, 256 ], "g_heads": [ 4, 4, 4 ], "g_mlp": [ 512, 512, 512 ], "d_initializer": "orthogonal", "d_enc_dim": [ 32, 64, 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,cutout", "fid_batch_size": 50, "gen_batch_size": 50, "loss": "nsl", "ema_decay": 0.999, "bcr": true, "cr_weight": 0.1, "dec_dim": [ 128, 64 ], "n_fid_real": 2500, "n_fid_gen": 2500, "plot_size": 5.2, "test_seed": 77 }