fix: argument

ghmagazine · Jul 15, 2024 · 1a1569f · 1a1569f
1 parent 3a495c3
commit 1a1569f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/abci/Llama-2-7b/llama-2-7b-tp2-pp2.sh b/scripts/abci/Llama-2-7b/llama-2-7b-tp2-pp2.sh
@@ -151,6 +151,7 @@ mpirun -np $NUM_GPUS \
   --recompute-activations \
   --recompute-granularity "selective" \
   --use-mpi \
+  --log-throughput \
   --wandb-name ${JOB_NAME} \
   --wandb-project "llm-book" \
   --wandb-entity "okoge"