ct_detection / mmdetection /tools /debug_train.sh
maverickrzw's picture
des
2402804
CONFIG='configs/cascade_rcnn/cascade-rcnn_x101-64x4d_fpn_1x_ct.py'
GPUS=8
WORKDIR='work_dirs/cascade-rcnn_x101-64x4d_fpn_1x_ct'
PORT=55565
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH
echo $PYTHONPATH
python3 -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
$(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:4} --work-dir $WORKDIR