-
Notifications
You must be signed in to change notification settings - Fork 6
/
oilseg_gan.sh
35 lines (32 loc) · 942 Bytes
/
oilseg_gan.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
#!/usr/bin/env bash
PRJROOT=${PWD}
rootpath=datasets
for exp in oilseg_gan
do
for dataset in syntheticSAR/salt_and_pepper syntheticSAR/multiplicative
do
for lr in 0.0002
do
#:<< !
python ${PRJROOT}/${exp}.py \
--mode train \
--output_dir ${PRJROOT}/checkpoints.lr.${lr}/${exp}/${dataset}/ \
--max_epochs 200 \
--input_dir ${rootpath}/${dataset}/train \
--which_direction AtoB \
--display_freq 100 \
--batch_size 1 \
--gan_weight 1.0 \
--lr ${lr}
#!
#: << !
python ${PRJROOT}/${exp}.py \
--mode test \
--output_dir ${PRJROOT}/results.lr.${lr}/${exp}/${dataset}/${f} \
--input_dir ${rootpath}/${dataset}/test \
--checkpoint ${PRJROOT}/checkpoints.lr.${lr}/${exp}/${dataset}/${f} \
--lr ${lr}
#!
done
done
done