fix training steps

ymcui · Jun 26, 2023 · c031387 · c031387
1 parent cc7d18e
commit c031387
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/training/run_sft.sh b/scripts/training/run_sft.sh
@@ -29,7 +29,7 @@ torchrun --nnodes 1 --nproc_per_node 1 run_clm_sft_with_peft.py \
     --do_eval \
     --seed $RANDOM \
     --fp16 \
-    --num_train_epochs 3 \
+    --num_train_epochs 1 \
     --lr_scheduler_type cosine \
     --learning_rate ${lr} \
     --warmup_ratio 0.03 \