File size: 337 Bytes
8c212a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/bin/bash
GPUS=$1
CONFIG=$2
WORK_DIR=$3
CHECKPOINT=$4
PORT=${PORT:-29500}
python -m torch.distributed.launch \
--nproc_per_node=${GPUS} \
--master_port=${PORT} \
./test.py ${CONFIG} \
--work_dir ${WORK_DIR} \
--checkpoint ${CHECKPOINT} \
--launcher="pytorch" \
${@:5}
|