调整日志输出

This commit is contained in:
997146918 2025-08-25 15:15:33 +08:00
parent 4bdf681cd4
commit 0fcce7fa46

View File

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