Spaces:
Running
Running
CONFIG=$1 | |
CHECKPOINT=$2 | |
GPUS=$3 | |
PORT=${PORT:-29510} | |
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \ | |
torchrun --nproc_per_node=$GPUS --master_port=$PORT \ | |
$(dirname "$0")/test.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4} | |