Update train_gpt_567B_1nodes.sh

e45e6793 · silencealiang · b9a97686 · e45e6793
Commit e45e6793 authored Apr 11, 2025 by silencealiang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/gpt3/train_gpt_567B_1nodes.sh examples/gpt3/train_gpt_567B_1nodes.sh +1 -1

No files found.
--- a/examples/gpt3/train_gpt_567B_1nodes.sh
+++ b/examples/gpt3/train_gpt_567B_1nodes.sh
@@ -109,7 +109,7 @@ TORCH_PROFIE_ARGS=(
    --profile-ranks 0 1 2 3 4 5 6 7
    --profile-step-start 3
    --profile-step-end 4
-    --profile-dir torch_prof_gpt_1nodes_tp2-pp1-ep8-ep_tp1-cp1
+    --profile-dir torch_prof_gpt_1nodes_tp2-pp1-ep4-ep_tp2-cp1
    --use-pytorch-profiler
 )