mirror of
https://github.com/open-mmlab/mmrazor.git
synced 2025-06-03 15:02:54 +08:00
f6d68dc73c
* update commands in README for 1.x * fix commands Co-authored-by: gaoyang07 <1546308416@qq.com>
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}
|