carry
|
088067d335
|
train: 更新模型训练功能和日志记录方式
- 修改训练目录结构，将检查点和日志分开保存
- 添加 TensorBoard 日志记录支持
- 移除自定义 LossCallback 类，简化训练流程
- 更新训练参数和回调机制，提高代码可读性
- 在 requirements.txt 中添加 tensorboardX 依赖
|
2025-04-14 16:19:37 +08:00 |
|
carry
|
9fb31c46c8
|
feat(train): 添加训练过程中的日志记录和 loss 可视化功能
- 新增 LossCallback 类，用于在训练过程中记录 loss 数据
- 在训练模型函数中添加回调，实现日志记录和 loss 可视化
- 优化训练过程中的输出信息，增加当前步数和 loss 值的打印
|
2025-04-14 15:18:14 +08:00 |
|
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
|
9784f2aed3
|
fix(tools): 修正__init__.py使得model.py正确导入
|
2025-04-12 01:16:07 +08:00 |
|
carry
|
8a9a080745
|
refactor(tools): 移除未使用的导入语句
移除了 tools/model.py 文件中未使用的 get_chat_template 导入语句。这个修改提高了代码的可读性和维护性。
|
2025-04-11 19:43:19 +08:00 |
|
carry
|
216bfe39ae
|
feat(tools): 添加格式化对话数据的函数
- 新增 formatting_prompts_func 函数，用于格式化对话数据
- 该函数将问题和答案组合成对话形式，并使用 tokenizer.apply_chat_template 进行格式化
- 更新 imports，添加了 unsloth.chat_templates 模块
|
2025-04-11 17:56:46 +08:00 |
|
carry
|
cbb3a09dd8
|
feat(tools): 添加模型名称获取函数
- 在 tools 目录下新增 model.py 文件
- 实现 get_model_name 函数，用于获取模型的名称
- 更新 tools/__init__.py，导入新的 get_model_name 函数
|
2025-04-10 22:05:04 +08:00 |
|