File size: 364 Bytes
2402804 |
1 2 3 4 5 6 7 8 9 10 11 |
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
|