Skip to content

Latest commit

 

History

History
23 lines (16 loc) · 596 Bytes

README.md

File metadata and controls

23 lines (16 loc) · 596 Bytes

Run Train:

$ python train.py --data_dir data --scale_factor 4 --patch_size 48 --batch_size 16 --num_epochs 5 --lr 1e-4 --num_workers 4 --device cuda --save_dir results --save_interval 2

Run Test:

$ python test.py --data_dir data --scale_factor 4 --patch_size 48 --batch_size 32 --num_workers 4 --device cuda --save_dir output --model_dir results/run_0/

Run Tensorboard:

$ tensorboard --logdir=results/ --port=2171

Evaluate Results:

$ python utils/evaluate.py --stitched_path stitched_output --hr_path data/DIV2K_train_HR