-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathtrain_cnn.sh
21 lines (15 loc) · 1.25 KB
/
train_cnn.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# Colorization
python code/train.py --experiment col_bedroom --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
python code/train.py --experiment col_church --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
# Blind denoising
python code/train.py --experiment denblind_bedroom --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
python code/train.py --experiment denblind_church --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
python code/train.py --experiment denblind_conference --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
# Inpainting CROP (64x64 center crop)
python code/train.py --experiment inpcrop_bedroom --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
python code/train.py --experiment inpcrop_church --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
python code/train.py --experiment inpcrop_conference --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
# Inpainting VAR (variable randomized mask)
python code/train.py --experiment inpvar_bedroom --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
python code/train.py --experiment inpvar_church --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5
python code/train.py --experiment inpvar_conference --lr 0.01 --batch_size 8 --backbone D --phi_weight 1e-5