add train epochs

2023-12-29 15:15:09 +08:00 · 2023-12-29 15:15:09 +08:00 · d4048deebf
parent e1b760b706
commit d4048deebf
1 changed files with 1 additions and 0 deletions
--- a/finetune/finetune-lora.py
+++ b/finetune/finetune-lora.py
@ -47,6 +47,7 @@ trainer = Trainer(
    model=lora_model,
    train_dataset=train_ds,
    args=TrainingArguments(
+        num_train_epochs=1,
        per_device_train_batch_size=4, 
        gradient_accumulation_steps=4,
        warmup_steps=3,