From 44aced819512fa7c07099a9f0831ac3bd157d4d4 Mon Sep 17 00:00:00 2001 From: inter Date: Sun, 21 Sep 2025 20:19:36 +0800 Subject: [PATCH] Add File --- tools/scripts/dist_test.sh | 7 +++++++ 1 file changed, 7 insertions(+) create mode 100644 tools/scripts/dist_test.sh 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}