| bash ./prepare.sh | |
| lst=0 | |
| for i in {0..3}; | |
| do | |
| if nvidia-smi -i $i | grep -q "python"; then | |
| : | |
| else | |
| lst=$i | |
| break | |
| fi | |
| done | |
| #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 | |
| CUDA_VISIBLE_DEVICES=$lst python main.py --n_GPUs 1 --lr 1e-5 --batch_size 16 --save_models --epoch 100 \ | |
| --model RAFTNETS --scale 50 --patch_size 48 --reset --save RAFTS_N50 --n_feats 64 --data_train DIV2K \ | |
| --n_resblocks 12 --recurrence 3 |