You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
- #!/usr/bin/env bash
-
- CONFIG=$1
- GPUS=$2
- PORT=${PORT:-29501}
-
- PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
- python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
- $(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3}
|