Update dist_test.sh ——torchrun

7e866c8a · dcuai · 458370f6 · 7e866c8a
Commit 7e866c8a authored Nov 02, 2024 by dcuai
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/dist_test.sh tools/dist_test.sh +1 -1

No files found.
--- a/tools/dist_test.sh
+++ b/tools/dist_test.sh
@@ -9,7 +9,7 @@ PORT=${PORT:-29500}
 MASTER_ADDR=${MASTER_ADDR:-"127.0.0.1"}

 PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
-python -m torch.distributed.launch \
+torchrun \
    --nnodes=$NNODES \
    --node_rank=$NODE_RANK \
    --master_addr=$MASTER_ADDR \