| 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 | |
| CUDA_VISIBLE_DEVICES=$lst python main.py --n_GPUs 1 --rgb_range 1 --reset --save_models --lr 1e-4 --decay 200-400-600-800 \ | |
| --save_results --n_resblocks 10 --n_feats 256 --res_scale 0.1 --batch_size 16 --model RAFTNETS --scale 2 --recurrence 4 \ | |
| --patch_size 36 --save RAFTS_R4_x2 --data_train DIV2K --data_range "1-800/901-919" | |