CAR/code/train-4rec-raft-s-gpu1-bn.sh

#!/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 --batch_size 16 --lr 1e-4 --decay 200-400-600-800 \
--save_models --n_resblocks 20 --model RAFTNETS --scale 10 --patch_size 48 --save RAFTNETS_Q10_bn \
--n_feats 64 --data_train DIV2K --recurrence 4 --normalization batch \
--load RAFTNETS_Q10_bn --resume -1

# 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 RAFTNETS --scale 50 --patch_size 48 --reset \
# --save RAFTS_N50_R4_nobn --n_feats 64 --data_train DIV2K --recurrence 4 --normalization none