-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathrun_wgan.sh
executable file
·149 lines (67 loc) · 26.3 KB
/
run_wgan.sh
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
# python train_only_local_GAN.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_6" --print_freq 100 --disp_freq 100 --load_warpnet "./checkpoints/exp_1/ckpt_020.pt"
# --load_checkpoint "checkpoints/exp_2/ckpt_080.pt"
# python train_only_local_GAN.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_6_debug_grad" --print_freq 10 --disp_freq 10 --load_warpnet "./checkpoints/exp_1/ckpt_020.pt"
# --load_checkpoint "checkpoints/exp_2/ckpt_080.pt"
# python train_full_part_GAN.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_11" --print_freq 100 --disp_freq 100 --load_warpnet "./checkpoints/exp_1/ckpt_040.pt" --load_checkpoint './checkpoints/exp_7/ckpt_180.pt'
# python train_full_part_local_GAN.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_12_2" --print_freq 100 --disp_freq 100 --load_checkpoint './checkpoints/exp_12/ckpt_215.pt' --save_epoch_freq 5 --pd_L_l_w 0.5 --pd_R_l_w 0.5 --pd_N_l_w 0.5 --pd_M_l_w 0.5
# python train_full_part_local_GAN.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_13" --print_freq 100 --disp_freq 100 --load_checkpoint './checkpoints/exp_12_2/ckpt_220.pt' --save_epoch_freq 10 --pd_L_l_w 1.5 --pd_R_l_w 1.5 --pd_N_l_w 1.5 --pd_M_l_w 1.5 --gd_l_w 2 --ld_l_w 1
# python train_full_part_local_GAN.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_13_2" --print_freq 100 --disp_freq 100 --load_checkpoint './checkpoints/exp_13/ckpt_240.pt' --save_epoch_freq 5 --pd_L_l_w 3 --pd_R_l_w 3 --pd_N_l_w 3 --pd_M_l_w 3 --gd_l_w 10 --ld_l_w 5
# python train_full_part_local_GAN.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_13_3" --print_freq 100 --disp_freq 100 --load_checkpoint './checkpoints/exp_13/ckpt_240.pt' --save_epoch_freq 1 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 2 --ld_l_w 1 --kind "weaker_1"
# python train_full_part_local_GAN_cond.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_16" --print_freq 100 --disp_freq 100 --save_epoch_freq 10 --pd_L_l_w 100 --pd_R_l_w 100 --pd_N_l_w 100 --pd_M_l_w 100 --gd_l_w 200 --ld_l_w 100 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --load_checkpoint './checkpoints/exp_13_3/ckpt_245.pt'
# python train_full_part_local_GAN_cond.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_17" --print_freq 100 --disp_freq 100 --save_epoch_freq 10 --pd_L_l_w 100 --pd_R_l_w 100 --pd_N_l_w 100 --pd_M_l_w 100 --gd_l_w 1 --ld_l_w 0.5 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --load_checkpoint './checkpoints/exp_16/ckpt_260.pt'
# dataset SP shrink parts set to 0.8
# python train_full_part_local_GAN_cond_SP_LR.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_18" --print_freq 100 --disp_freq 100 --save_epoch_freq 10 --pd_L_l_w 100 --pd_R_l_w 100 --pd_N_l_w 100 --pd_M_l_w 100 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 10 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --load_checkpoint './checkpoints/exp_17/ckpt_300.pt'
# dataset SP shrink parts set to 1.2
# python train_full_part_local_GAN_cond_SP_LR.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_19" --print_freq 100 --disp_freq 100 --save_epoch_freq 5 --pd_L_l_w 1 --pd_R_l_w 1 --pd_N_l_w 1 --pd_M_l_w 1 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --load_checkpoint './checkpoints/exp_18/ckpt_310.pt' --parts_expand 1.2
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_43" --print_freq 10 --disp_freq 10 --save_epoch_freq 10 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --perp_l_w 0 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN --load_warpnet "./checkpoints/exp_1/ckpt_040.pt"
# --adam
# --load_checkpoint './checkpoints/exp_19/ckpt_465.pt'
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_43_just_wgan" --print_freq 100 --disp_freq 100 --save_epoch_freq 10 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --perp_l_w 0 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN --load_warpnet "./checkpoints/exp_1/ckpt_040.pt"
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_44_only_part_lr=2e-5" --print_freq 100 --disp_freq 100 --save_epoch_freq 1 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --perp_l_w 0 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --load_warpnet "./checkpoints/exp_1/ckpt_040.pt" --lr 2e-5
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_only_mse_debug_HPC" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --perp_l_w 0 --kind "original" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --load_warpnet "./checkpoints/exp_1/ckpt_040.pt" --jpeg_last --deconv_kind "subpixel" --skip_train_D --lr 4e-3 --debug --num_workers 4 --batch_size 8
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_subpixel_lr_dbg" --print_freq 10 --disp_freq 10 --save_epoch_freq 1 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --perp_l_w 0 --kind "original" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --load_warpnet "./checkpoints/exp_1/ckpt_040.pt" --lr 2e-4 --jpeg_last --deconv_kind "subpixel" --skip_train_D --debug --lr 4e-3
# --mse_l_w 2
# --debug
# --deconv_kind "subpixel"
# weaker_1 subpixel finetune only mse, no jpeg_last
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_subpixel_ft_weaker1" --print_freq 10 --disp_freq 10 --save_epoch_freq 1 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --perp_l_w 0 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --skip_train_D --debug --lr 4e-3 --load_checkpoint "./checkpoints/exp_45_subpixel_lr_dbg/ckpt_050.pt"
# exp_45_test_vgg_[conv3]
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_test_vgg_[conv3]" --print_freq 10 --disp_freq 10 --save_epoch_freq 1 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --skip_train_D --lr 4e-3 --load_checkpoint "./checkpoints/exp_45_subpixel_ft_weaker1/ckpt_066.pt" --mse_l_w 0.1 --perp_l_w 0.001
# --debug
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_test_vgg_[conv3]_2" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --skip_train_D --lr 2e-4 --load_checkpoint "./checkpoints/exp_45_subpixel_ft_weaker1/ckpt_066.pt" --mse_l_w 0.1 --perp_l_w 0.01
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_test_vgg_[conv3]_3" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --skip_train_D --lr 2e-4 --load_checkpoint "./checkpoints/exp_45_subpixel_ft_weaker1/ckpt_066.pt" --mse_l_w 0.1 --perp_l_w 0.1
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_test_vgg_[conv4]_3_dbg" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --skip_train_D --lr 2e-4 --load_checkpoint "./checkpoints/exp_45_subpixel_ft_weaker1/ckpt_066.pt" --mse_l_w 0 --perp_l_w 0.1 --vgg_conv_X 4
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_test_vgg_[conv4]_4" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --skip_train_D --lr 2e-4 --load_checkpoint "./checkpoints/exp_45_subpixel_ft_weaker1/ckpt_066.pt" --mse_l_w 0.1 --perp_l_w 0.001 --vgg_conv_X 4
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_45_test_vgg_[conv4]_5" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --skip_train_D --lr 2e-4 --load_checkpoint "./checkpoints/exp_45_test_vgg_[conv4]_4/ckpt_082.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_46_add_part" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 2 --pd_R_l_w 2 --pd_N_l_w 2 --pd_M_l_w 2 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_45_test_vgg_[conv4]_5/ckpt_084.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_46_add_part_w=10" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part/ckpt_116.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_46_add_part_w=10_grad" --print_freq 100 --disp_freq 100 --save_epoch_freq 2 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=10/ckpt_132.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --debug --no_prewarm_D --manual_seed 1996 --pd_L_l_w 0 --pd_R_l_w 0 --pd_N_l_w 0 --pd_M_l_w 0
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_46_add_part_w=2000" --print_freq 100 --disp_freq 100 --save_epoch_freq 5 --pd_L_l_w 2000 --pd_R_l_w 2000 --pd_N_l_w 2000 --pd_M_l_w 2000 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=10/ckpt_132.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --no_prewarm_D
# remove no_prewarm_D
# more warm(warm_interval:100)
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_46_add_part_w=1_noRecLoss_morewarm" --print_freq 100 --disp_freq 100 --save_epoch_freq 5 --pd_L_l_w 1 --pd_R_l_w 1 --pd_N_l_w 1 --pd_M_l_w 1 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=10/ckpt_132.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --mse_l_w 0 --perp_l_w 0 --warm_interval 100
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_46_add_part_w=1_RecLoss_morewarm" --print_freq 100 --disp_freq 100 --save_epoch_freq 5 --pd_L_l_w 1 --pd_R_l_w 1 --pd_N_l_w 1 --pd_M_l_w 1 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=1_noRecLoss_morewarm/ckpt_220.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 100
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_1" --print_freq 100 --disp_freq 100 --save_epoch_freq 5 --pd_L_l_w 1 --pd_R_l_w 0 --pd_N_l_w 0 --pd_M_l_w 0 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=1_RecLoss_morewarm/ckpt_885.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 100 --no_rec_loss --prewarm_len 10
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_1_dbg" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 100 --pd_R_l_w 100 --pd_N_l_w 100 --pd_M_l_w 100 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=1_RecLoss_morewarm/ckpt_885.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 100 --prewarm_len 10
# --debug
# --no_rec_loss
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_2" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 100 --pd_R_l_w 100 --pd_N_l_w 100 --pd_M_l_w 100 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=1_RecLoss_morewarm/ckpt_885.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 100 --prewarm_len 10 --debug --num_workers 0
# remove perp loss from rec loss
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_2" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 100 --pd_R_l_w 100 --pd_N_l_w 100 --pd_M_l_w 100 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_46_add_part_w=1_RecLoss_morewarm/ckpt_885.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 100 --prewarm_len 10 --warm_Diters 10
# warm interval 减小为50， 每次warm up 10
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_3_dataparallelDbg" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 100 --pd_R_l_w 100 --pd_N_l_w 100 --pd_M_l_w 100 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_47_2/ckpt_894.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 50 --prewarm_len 10 --warm_Diters 10 --use_mult_gpus
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_5" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_47_3_dataparallelDbg/ckpt_934.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 50 --prewarm_len 10 --warm_Diters 100 --use_mult_gpus --debug --manual_seed 4057
# batchsize=14 per gpu(4)
# tot batchsize=16 gpu(2)
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_4" --print_freq 50 --disp_freq 50 --save_epoch_freq 2 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-5 --load_checkpoint "./checkpoints/exp_47_3_dataparallelDbg/ckpt_934.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 50 --prewarm_len 10 --warm_Diters 100 --use_mult_gpus
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_4_dbg" --print_freq 50 --disp_freq 50 --save_epoch_freq 2 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 0.5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-5 --load_checkpoint "./checkpoints/exp_47_3_dataparallelDbg/ckpt_934.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 50 --prewarm_len 10 --warm_Diters 100 --use_mult_gpus --debug
# --debug
# --no_rec_loss
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_6_1" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 5 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 1e-4 --load_checkpoint "./checkpoints/exp_47_3_dataparallelDbg/ckpt_934.pt" --mse_l_w 0.3 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 50 --prewarm_len 10 --warm_Diters 50 --use_mult_gpus --manual_seed 4057
# --gp_lambda 10
# ld_l_w 10, lr 2e-5, gp_lambda 100, mse_l_w 0.3
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_6_2_lambda=100" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 10 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-5 --load_checkpoint "./checkpoints/exp_47_3_dataparallelDbg/ckpt_934.pt" --mse_l_w 0.3 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 500 --prewarm_len 10 --warm_Diters 100 --use_mult_gpus --gp_lambda 100
# python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_6_2_lambda=100_2" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 20 --pd_R_l_w 20 --pd_N_l_w 20 --pd_M_l_w 20 --gd_l_w 1 --ld_l_w 10 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_47_6_2_lambda=100/ckpt_963.pt" --mse_l_w 0.3 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 500 --prewarm_len 10 --warm_Diters 100 --use_mult_gpus --gp_lambda 50
python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_6_2_lambda=100_3" --print_freq 1 --disp_freq 1 --save_epoch_freq 1 --pd_L_l_w 20 --pd_R_l_w 20 --pd_N_l_w 20 --pd_M_l_w 20 --gd_l_w 1 --ld_l_w 10 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_47_6_2_lambda=100/ckpt_963.pt" --mse_l_w 0.1 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 50 --prewarm_len 10 --warm_Diters 10 --use_mult_gpus --gp_lambda 100
# ld_l_w 10, lr 2e-5, gp_lambda 100, mse_l_w 0.3
python train_wgan.py --cuda --train_img_dir "./DataSets/Original/Train" --train_landmark_dir "./DataSets/Original/Landmark" --test_img_dir "./DataSets/Original/Test/testvgg" --test_landmark_dir "./DataSets/Original/Landmark" --max_epoch 1500 --exp_name "exp_47_6_2_lambda=100_symm" --print_freq 100 --disp_freq 100 --save_epoch_freq 1 --pd_L_l_w 10 --pd_R_l_w 10 --pd_N_l_w 10 --pd_M_l_w 10 --gd_l_w 1 --ld_l_w 10 --lr_l_w 1 --kind "weaker_1" --GD_cond 6 --PD_cond 6 --parts_expand 1.2 --use_WGAN_GP --deconv_kind "subpixel" --lr 2e-4 --load_checkpoint "./checkpoints/exp_47_3_dataparallelDbg/ckpt_934.pt" --mse_l_w 0.3 --perp_l_w 0.005 --vgg_conv_X 4 --warm_interval 500 --prewarm_len 10 --warm_Diters 100 --use_mult_gpus --gp_lambda 100 --load_warpnet "./checkpoints/exp_42_l2_tot_loss/ckpt_100.pt"