Update dist_train.sh

a9be62f0 · dcuai · 24240e28 · a9be62f0
Commit a9be62f0 authored Sep 14, 2024 by dcuai
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tools/dist_train.sh tools/dist_train.sh +2 -2

No files found.
--- a/tools/dist_train.sh
+++ b/tools/dist_train.sh
@@ -16,5 +16,5 @@ python -m torch.distributed.launch \
    --nproc_per_node=$GPUS \
    --master_port=$PORT \
    $(dirname "$0")/train.py \
-    $CONFIG \
-    --launcher pytorch ${@:3}
+    $CONFIG 
+