Add File

2025-09-21 20:19:36 +08:00
parent b4aefdc5f8
commit 44aced8195
1 changed files with 7 additions and 0 deletions
--- a/tools/scripts/dist_test.sh
+++ 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}