diff --git a/tools/scripts/dist_test.sh b/tools/scripts/dist_test.sh new file mode 100644 index 0000000..33460ff --- /dev/null +++ b/tools/scripts/dist_test.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash + +set -x +NGPUS=$1 +PY_ARGS=${@:2} + +python -m torch.distributed.launch --nproc_per_node=${NGPUS} test.py --launcher pytorch ${PY_ARGS}