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', # 训练参数