forked from aiff22/PyNET
-
Notifications
You must be signed in to change notification settings - Fork 5
/
train.sh
21 lines (19 loc) · 1.9 KB
/
train.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
python train_model_keras.py level=3 restore_iter=0 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="vgg+ssim"
cp 73/models/model.3.best.h5 73/models/model.4.9999.h5
python train_model_keras.py level=3 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="mse+ssim"
cp 73/models/model.3.best.h5 73/models/model.3.9999.h5
python train_model_keras.py level=2 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="vgg+ssim"
cp 73/models/model.2.best.h5 73/models/model.3.9999.h5
python train_model_keras.py level=2 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="mse+ssim"
cp 73/models/model.2.best.h5 73/models/model.2.9999.h5
python train_model_keras.py level=1 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="vgg+ssim"
cp 73/models/model.1.best.h5 73/models/model.2.9999.h5
python train_model_keras.py level=1 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="mse+ssim"
cp 73/models/model.1.best.h5 73/models/model.2.9999.h5
# fine tuning
#CUDA_VISIBLE_DEVICES=1 python train_model_keras.py level=1 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="ssim"
#cp 73/models/model.1.best.h5 73/models/model.2.9999.h5
#python train_model_keras.py level=1 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="vgg+ssim"
#cp 73/models/model.1.best.h5 73/models/model.2.9999.h5
#python train_model_keras.py level=1 restore_iter=9999 dataset_dir=raw_images_full/ train_size=95000 dir_prefix=73/ model_path=73/model.py loss_fn="mse+ssim"
#cp 73/models/model.1.best.h5 73/models/model.2.9999.h5