File size: 499 Bytes
d6ec83b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
#!/bin/bash
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"
|