调整日志输出
This commit is contained in:
parent
4bdf681cd4
commit
0fcce7fa46
@ -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',
|
||||
|
||||
# 训练参数
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user