local_train.sh

export CUDA_VISIBLE_DEVICES=0,1
export PYTHONPATH=$PYTHONPATH:./

gpus=(${CUDA_VISIBLE_DEVICES//,/ })
gpu_num=${#gpus[@]}
echo "number of gpus: "${gpu_num}

config=projects/configs/$1.py

if [ ${gpu_num} -gt 1 ]
then
    bash ./tools/dist_train.sh \
        ${config} \
        ${gpu_num} \
        --work-dir=work_dirs/$1
else
    python ./tools/train.py \
        ${config}
fi

# python tools/train.py projects/configs/sparse4dv3_temporal_r50_1x8_bs6_256x704.py
# nohup bash local_train.sh sparse4dv3_temporal_r50_1x8_bs6_256x704 > train.log 2>&1 & 3949
# nohup python train.py > train.log 2>&1 &