LadaGAN / bedroom_128 /config.json
milmor's picture
Upload 4 files
a613a93 verified
raw
history blame
938 Bytes
{
"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
}