 carry
|
4f09823123
|
refactor(tools): 优化 train_model 函数定义
- 添加类型注解，提高代码可读性和维护性
- 使用多行格式定义函数参数，提升代码格式美观
|
2025-04-14 14:28:36 +08:00 |
|
|
carry
|
1a2ca3e244
|
refactor(train): 重构训练功能并移至新模块
- 将训练逻辑从 train_page.py 移至 tools/model.py
- 新增 train_model 函数，包含完整的训练流程
- 更新 train_page.py 中的回调函数，使用新的训练函数
- 移除了 train_page.py 中未使用的导入
|
2025-04-14 14:17:04 +08:00 |
|
|
carry
|
bb1d8fbd38
|
feat(train_page): 添加训练 Loss 曲线显示功能
- 在训练页面添加了 Loss 曲线图表
- 实现了 GradioLossCallback 类用于记录训练过程中的 Loss 数据
- 修改了训练函数，通过回调函数收集 Loss 信息并更新图表
- 优化了训练函数的返回值结构，支持同时返回文本日志和 Loss 数据
|
2025-04-13 21:49:43 +08:00 |
|
|
carry
|
4558929c52
|
fix: 调整了import的顺序，让unsloth最先import以提高性能
|
2025-04-13 21:35:47 +08:00 |
|
|
carry
|
0722748997
|
feat(train_page): 添加 LoRA 秩动态输入功能
- 在训练页面新增 LoRA 秩输入框，使用户可以动态设置 LoRA 秩
- 更新训练模型函数，添加 LoRA 秩参数并将其用于模型配置
- 保留原有功能，仅增加 LoRA 秩相关配置
|
2025-04-13 21:12:02 +08:00 |
|
|
carry
|
e08f0059bb
|
feat(train_page): 优化训练过程以专注于响应生成
- 引入 train_on_responses_only 函数，用于优化训练过程
- 设置 instruction_part 和 response_part 参数，以适应特定的对话格式
- 此修改旨在提高模型在生成响应方面的性能和效率
|
2025-04-13 21:05:14 +08:00 |
|