hyliu's picture
Upload folder using huggingface_hub
2541ce6 verified
#!/bin/bash
# PANET Train
#python main.py --n_GPUs 1 --lr 1e-5 --batch_size 16 --save_models --epoch 100 --model PANET --scale 50 --patch_size 48 --reset --save MDSR_att_N50 --n_feats 64 --data_train DIV2K --chop
#PANET Test
python main.py --model PANET --save_results --n_GPUs 1 --chop --data_test Kodak24+CBSD68+Urban100 --scale 10 --pre_train ../model_N10.pt --test_only