This commit is contained in:
2025-09-21 20:19:36 +08:00
parent b4aefdc5f8
commit 44aced8195

View File

@@ -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}