forked from eleGAN23/HI2I
-
Notifications
You must be signed in to change notification settings - Fork 3
/
config_n=3.json
55 lines (55 loc) · 1.44 KB
/
config_n=3.json
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
53
54
55
{
"experiment_name":"phm_3dense",
"gpu_num": 0,
"dataset":"celeba_hq",
"num_domains": 2,
"img_size" : 129,
"batch_size" : 1,
"val_batch_size" : 1,
"randcrop_prob" : 0.5,
"num_workers" : 2,
"train_img_dir" : "./Datasets/afhq/train",
"val_img_dir" : "./Datasets/afhq/val",
"eval_dir": "./expr/eval",
"result_dir": "./expr/results",
"checkpoint_dir": "./expr/checkpoints",
"wing_path": "./expr/checkpoints/wing.ckpt",
"sample_dir":"./expr/samples",
"src_dir": "./assets/representative/celeba_hq/src",
"ref_dir": "./assets/representative/celeba_hq/ref",
"inp_dir": "./assets/representative/custom/female",
"out_dir": "./assets/representative/afhq/src/dog",
"w_hpf":1 ,
"beta1" :0,
"beta2":0.99,
"latent_dim": 16,
"hidden_dim":512,
"style_dim": 64,
"mode" : "sample",
"lr" : 1e-4,
"f_lr": 1e-6,
"weight_decay": 1e-4,
"lambda_ds":1,
"lambda_reg":1,
"lambda_cyc":1,
"lambda_sty":1,
"ds_iter":100000,
"total_iters":100000,
"resume_iter":100000,
"print_every":10,
"sample_every":1,
"save_every" :1000,
"eval_every":100000,
"num_outs_per_domain":10,
"seed":777,
"layer_norm": "False",
"quat_inst_norm": "False",
"quat_max_pool" : "False",
"qsngan_layers" : "False",
"htorch_layers":"False",
"real": "False",
"phm":"True",
"N":3,
"last_dense":"True",
"disc_dense": 4
}