mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
21 lines
443 B
Bash
Executable File
21 lines
443 B
Bash
Executable File
|
|
#!/usr/bin/env bash
|
|
|
|
CONFIG=$1
|
|
GPUS=$2
|
|
NNODES=${NNODES:-1}
|
|
NODE_RANK=${NODE_RANK:-0}
|
|
PORT=${PORT:-29500}
|
|
MASTER_ADDR=${MASTER_ADDR:-"127.0.0.1"}
|
|
|
|
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
|
|
python -m torch.distributed.launch \
|
|
--nnodes=$NNODES \
|
|
--node_rank=$NODE_RANK \
|
|
--master_addr=$MASTER_ADDR \
|
|
--nproc_per_node=$GPUS \
|
|
--master_port=$PORT \
|
|
$(dirname "$0")/train.py \
|
|
$CONFIG \
|
|
--launcher pytorch ${@:3}
|