mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
20 lines
442 B
Bash
20 lines
442 B
Bash
#!/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}
|