-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_main.sh
37 lines (34 loc) · 873 Bytes
/
run_main.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
# !bash ./run_main.sh
#python3 './main.py' \
# --mode 'train' \
# --lr 2e-4 \
# --batch_size 10 \
# --num_epoch 300 \
# --ny 256 \
# --nx 256 \
# --nch 3 \
# --nker 64 \
# --wgt 1e2 \
# --network 'pix2pix' \
# --data_dir './../../datasets/facaes' \
# --ckpt_dir './checkpoint' \
# --log_dir './log' \
# --result_dir './result'
#
python3 './main.py' \
--mode 'train' \
--train_continue 'on' \
--lr 2e-4 \
--batch_size 4 \
--num_epoch 100 \
--ny 256 \
--nx 256 \
--nch 3 \
--nker 64 \
--wgt_cycle 1e2 \
--wgt_ident 5e-1 \
--network 'cyclegan' \
--data_dir './spring_5000' \
--ckpt_dir './checkpoint' \
--log_dir './log' \
--result_dir './result'