From 0fcce7fa461a07e6bed99a87dfed7a6c3330559c Mon Sep 17 00:00:00 2001
From: 997146918 <997146918@qq.com>
Date: Mon, 25 Aug 2025 15:15:33 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E6=97=A5=E5=BF=97=E8=BE=93?=
 =?UTF-8?q?=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 AITrain/iterative_lora_training.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/AITrain/iterative_lora_training.py b/AITrain/iterative_lora_training.py
index e12277d..6d787ce 100644
--- a/AITrain/iterative_lora_training.py
+++ b/AITrain/iterative_lora_training.py
@@ -5,6 +5,7 @@
 只保留核心的数据训练模型功能
 """
 
+import time
 import json
 import torch
 from transformers import (
@@ -223,11 +224,11 @@ class IterativeLoRATrainer:
         tokenizer.save_pretrained(final_output_dir)
         
         print(f"训练完成，模型保存到: {self.output_path}")
-
+        time.sleep(3)
 def main():
     config = {
         'base_model_path': '/mnt/g/Project02/AITrain/Qwen/Qwen3-4B',
-        'training_data_path': './training_data/high_quality_dialogues_20250823_1819.json',
+        'training_data_path': './training_data/high_quality_dialogues.json',
         'output_path': './output/iterative_lora_simple',
         
         # 训练参数