-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathmeta_train.sh
34 lines (34 loc) · 1.39 KB
/
meta_train.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
# train experiment
python3 meta_main.py --save_dir ../exp/meta_train/GOPRO/reblur_exp_attn_mprnet_gan_0.99_cycle_update_n_20_patch_gopro \
--reset True \
--log_file_name train.log \
--gpu_id 9 \
--random_seed 43 \
--num_workers 5 \
--dataset GOPRO \
--dataset_dir /disk1/psliu/datasets/benchMark/GOPRO_Large \
--batch_size 1 \
--meta True \
--meta_train True \
--reblur_lr 2.5e-6 \
--deblur_lr 2.5e-6 \
--use_inner_lr False \
--inner_lr 1e-6 \
--num_epochs 3 \
--save_every 1 \
--img_w 1280 \
--img_h 720 \
--input_w 256 \
--input_h 256 \
--reblur_model attn \
--deblur_model mprnet \
--n_frames 5 \
--gan True \
--gan_lr 1e-6 \
--gan_ratio 0.99 \
--cycle_update True \
--full_img_exp False \
--n_updates 20 \
--deblur_model_path weights/MPRNet/model_deblurring.pkl \
--reblur_model_path ../exp/train/GOPRO/reblur_exp_attn_zero2one_gan/model/reblur_01000.pt \
--gan_model_path ../exp/train/GOPRO/reblur_exp_attn_zero2one_gan/model/gan_01000.pt