File size: 801 Bytes
3ef0208 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
#!/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 --lr 1e-4 --batch_size 16 --n_resblocks 20 --save_models \
# --epoch 1000 --decay 200-400-600-800 --model RAFTNETSINGLE --scale 50 --patch_size 48 --reset \
# --save RAFTSINGLE_N50_R1_nobn --n_feats 64 --data_train DIV2K --recurrence 1 --normalization none
CUDA_VISIBLE_DEVICES=$lst python main.py --n_GPUs 1 --batch_size 16 --lr 1e-4 --decay 200-400-600-800 \
--save_models --n_resblocks 20 --model RAFTNETSINGLE --scale 10 --patch_size 48 --save RAFTSINGLE_Q10_bn \
--n_feats 64 --data_train DIV2K --recurrence 1 --normalization batch \
--load RAFTSINGLE_Q10_bn --resume -1 |