From d4048deebf3af9bbc232909337000a6b70a9ed69 Mon Sep 17 00:00:00 2001
From: wanghh2000 <wanghh2000@163.com>
Date: Fri, 29 Dec 2023 15:15:09 +0800
Subject: [PATCH] add train epochs

---
 finetune/finetune-lora.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/finetune/finetune-lora.py b/finetune/finetune-lora.py
index 8c7de04..5d3e0f9 100644
--- 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,