admin/Project02
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

调整日志输出

Browse Source
This commit is contained in:
997146918 2025-08-25 15:15:33 +08:00
parent 4bdf681cd4
commit 0fcce7fa46
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
AITrain/iterative_lora_training.py
View File

@ -5,6 +5,7 @@
只保留核心的数据训练模型功能 只保留核心的数据训练模型功能
""" """
import time
import json import json
import torch import torch
from transformers import ( from transformers import (
@ -223,11 +224,11 @@ class IterativeLoRATrainer:
tokenizer.save_pretrained(final_output_dir) tokenizer.save_pretrained(final_output_dir)
print(f"训练完成，模型保存到: {self.output_path}") print(f"训练完成，模型保存到: {self.output_path}")
time.sleep(3)
def main(): def main():
config = { config = {
'base_model_path': '/mnt/g/Project02/AITrain/Qwen/Qwen3-4B', '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', 'output_path': './output/iterative_lora_simple',
# 训练参数 # 训练参数