| CONFIG=configs/anymo_audio.py | |
| WORD_DIR=work_dirs/debug | |
| GPUS=1 | |
| NNODES=1 #${NNODES:-1} | |
| NODE_RANK=0 #${NODE_RANK:-0} | |
| PORT=${PORT:-29501} | |
| # MASTER_ADDR=${MASTER_ADDR:-"127.0.0.1"} | |
| MASTER_ADDR="127.0.0.1" | |
| export PORT=29501 | |
| export TORCH_DISTRIBUTED_DEBUG=INFO | |
| 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 \ | |
| --work-dir $WORD_DIR \ | |
| $CONFIG \ | |
| --launcher pytorch ${@:3} | |
| # sleep 1089 | |
| # python gpu.py | |