voc0712_retinanet_r101_baseline.sh

#!/usr/bin/env bash

# run by PORT=4444 bash faster_rcnn_resnet50_baseline.sh 0,1,2,3 
GPUS=$1
PORT=${PORT}

CUDA_VISIBLE_DEVICES=$GPUS python -m torch.distributed.launch --master_port $PORT --nproc_per_node=4 --use_env \
	src/train.py --dataset voc0712 --model retinanet_resnet101_fpn --epochs 24 --lr-steps 18 22 --lr 0.005 --lr-warmup 1000 \
	--backbone-pretrained https://download.pytorch.org/models/resnet101-5d3b4d8f.pth