File size: 881 Bytes
b41e94c 0d0a2a5 0c496d9 b41e94c 0d0a2a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
{
"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
}
|