-
Notifications
You must be signed in to change notification settings - Fork 3
/
start_train.py
76 lines (76 loc) · 860 Bytes
/
start_train.py
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
import os
os.system('python train.py --dataroot \
./datasets/SICE/train \
--no_dropout \
--name \
enlightening \
--dataset_mode \
pair \
--which_model_netG \
sid_unet_resize \
--which_model_netD \
no_norm_4 \
--n_layers_D \
5 \
--n_layers_patchD \
4 \
--fineSize \
320 \
--patchSize \
32 \
--skip \
1 \
--batchSize \
1 \
--use_norm \
1 \
--use_wgan \
0 \
--use_ragan \
--hybrid_loss \
--times_residual \
--instance_norm \
0 \
--vgg \
1 \
--vgg_choose \
relu5_1 \
--gpu_ids \
2 \
--display_port \
8097 \
--lr \
0.0002 \
--continue_train \
0 \
--continue_epoch \
0 \
--fullinput \
0 \
--patchD_3 \
0 \
--hasglobal \
1 \
--mse_weight \
1 \
--vgg_weight \
0.1 \
--gan_weight \
0.1 \
--contract_weight \
0.1 \
--global_local_rate \
10 \
--niter \
100 \
--niter_decay \
100 \
--ssim_loss \
0.1 \
--self_attention \
--model \
double \
--retina \
1 \
'
)