Merge pull request hiyouga#3578 from pha123661/main

Fix badam example argument
xin-li-67 · May 5, 2024 · a34f526 · a34f526
2 parents 845d5ac + eeb415f
commit a34f526
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/extras/badam/sft.sh b/examples/extras/badam/sft.sh
@@ -10,7 +10,7 @@ CUDA_VISIBLE_DEVICES=0 llamafactory-cli train \
     --finetuning_type full \
     --use_badam \
     --badam_switch_mode descending \
-    --badam_switch_block_every 50 \
+    --badam_switch_interval 50 \
     --badam_verbose 2 \
     --output_dir ../../../saves/LLaMA2-7B/badam/sft \
     --overwrite_cache \