Commit Graph

110 Commits

Author SHA1 Message Date
ViperEkura 144b9598ad feat(model): 添加 Linear 和 Embedding 模块的自定义参数初始化支持 2025-10-31 22:43:12 +08:00
ViperEkura 877669b799 feat(inference): 添加generate_loop方法并优化KVCacheManager初始化 2025-10-31 21:15:15 +08:00
ViperEkura e86328b753 fix(tokenizer): 修复stop_ids属性返回错误的token ID列表 2025-10-31 19:19:38 +08:00
ViperEkura 5d3799b715 refactor(data): 修改变量命名方式 2025-10-30 16:32:25 +08:00
ViperEkura 6a3135f401 fix(data_util): 修复数据集索引计算逻辑并提取通用方法 2025-10-29 20:58:33 +08:00
ViperEkura 12850d403c fix(config): 修改Checkpoint类中tokenizer和config字段的默认值初始化方式 2025-10-29 13:24:20 +08:00
ViperEkura d94fc5a87a feat(data, inference): 使用chatML格式 2025-10-29 12:02:43 +08:00
ViperEkura 38b2725cd1 feat(KVCacheManager): 优化KV缓存结构为元组形式以提升性能 2025-10-29 12:01:28 +08:00
ViperEkura bc5ef72001 fix(config): 修正 SGDRScheduleConfig 类名拼写错误 2025-10-20 18:21:46 +08:00
ViperEkura 0db046f8d9 feat(khaosz/trainer): 更新梯度裁剪回调 2025-10-20 13:30:26 +08:00
ViperEkura 05b012820b refactor(khaosz): 重构模块导出结构并重命名主模块文件 2025-10-20 13:07:02 +08:00
ViperEkura e72e244df6 feat(inference): 实现采样策略并优化生成器逻辑 2025-10-20 13:00:41 +08:00
ViperEkura 98efca7b9d feat(trainer): 添加训练起始轮次和批次配置支持 2025-10-19 21:47:10 +08:00
ViperEkura 622982364b fix(trainer): 修复检查点加载逻辑 2025-10-18 21:45:23 +08:00
ViperEkura b67bc9865d refactor(trainer): 重构学习率调度器实现并分离配置与工厂逻辑 2025-10-18 16:42:37 +08:00
ViperEkura c51b203fde refactor(khaosz): 重构项目结构 2025-10-18 13:56:59 +08:00
ViperEkura 8434c19923 fix(khaosz/trainer): 修复数据获取中的索引范围错误和参数传递问题 2025-10-09 19:53:52 +08:00
ViperEkura efbe3de9d3 fix(khaosz/trainer/data_util): 修复数据集索引范围错误 2025-10-07 20:04:45 +08:00
ViperEkura 12793bc2d3 feat(khaosz/trainer): 新增梯度统计工具函数并重构训练回调机制 2025-10-07 13:03:32 +08:00
ViperEkura 0764cb8296 fix(khaosz/trainer/train_callback): 修复基类函数命名错误 2025-10-07 11:43:51 +08:00
ViperEkura 57cd7b921e feat(khaosz/trainer): 改进训练循环中的损失归一化处理 2025-10-06 20:17:47 +08:00
ViperEkura c1bf22b6ec refactor(khaosz/trainer): 使用 TrainContext 替代 kwargs 传递训练上下文 2025-10-06 20:12:08 +08:00
ViperEkura f9b6331ad7 refactor(khaosz/core/parameter): 修改参数名称 2025-10-06 20:11:46 +08:00
ViperEkura 183f481692 build(khaosz): 更新版本号至1.3.0 2025-10-06 17:12:12 +08:00
ViperEkura 4ffa7454f2 feat(strategy): 支持模型输入可调用对象并优化损失计算 2025-10-06 17:08:56 +08:00
ViperEkura fc98d9b7e6 refactor(khaosz/trainer): 移除未使用的导入模块 2025-10-04 21:45:53 +08:00
ViperEkura 2ccd7bd583 refactor(khaosz/trainer): 重构训练器模块结构以提升可维护性 2025-10-04 21:31:15 +08:00
ViperEkura 465a1a9373 refactor(khaosz/tainer): 修改设备参数传递发生阶段 2025-10-04 12:12:21 +08:00
ViperEkura 240ee00221 feat(khaosz/trainer): 引入 TrainContext 和 TrainContextBuilder 优化训练上下文管理 2025-10-03 22:42:11 +08:00
ViperEkura 9a452dd34e fix(khaosz/trainer/data_util.py): 修复 RandomSampler 中迭代计数器位置错误 2025-10-03 22:08:28 +08:00
ViperEkura 83c08cfbb9 fix(trainer): 修复随机采样器迭代重置问题 2025-10-02 14:22:38 +08:00
ViperEkura e43a5b9b66 fix(khaosz/trainer/data_util.py): 修复 build_loss_mask 函数中使用错误的输入张量 2025-10-02 11:55:51 +08:00
ViperEkura 753e493d6c docs(khaosz/trainer): 更新回调函数的文档字符串格式 2025-09-30 20:55:24 +08:00
ViperEkura 5a356d66e1 refactor(khaosz/trainer/data_util): 重构mask函数 2025-09-30 20:22:12 +08:00
ViperEkura 78e5dbb3be build(khaosz): 更新版本号至1.2.2 2025-09-30 18:04:13 +08:00
ViperEkura 288e2c3da6 fix(khaosz/trainer): 调整训练循环中回调调用顺序并增强异常日志记录 2025-09-30 17:57:55 +08:00
ViperEkura 315ce1990a feat(khaosz/trainer): 优化训练器回调机制与数据采样逻辑 2025-09-30 16:33:18 +08:00
ViperEkura e0e9942e4a feat(khaosz/core/parameter): 添加采样器状态的保存与加载功能 2025-09-29 19:49:35 +08:00
ViperEkura 198c1ac55c refactor(khaosz/trainer): 重构训练器模块的导入路径和文件命名 2025-09-29 19:35:17 +08:00
ViperEkura e467420475 fix(khaosz/trainer): 修正训练步数计算逻辑 2025-09-29 19:05:26 +08:00
ViperEkura 6d5176a11c feat(khaosz/trainer): 改进调度器配置验证和加载逻辑 2025-09-29 17:17:45 +08:00
ViperEkura bdda1cc35a feat(khaosz/core/tokenizer): 添加 user_id 和 system_id 属性 2025-09-29 13:47:37 +08:00
ViperEkura 89211c16f6 fix(khaosz/trainer): 将保存检查点逻辑移至CheckpointCallback 2025-09-29 13:38:46 +08:00
ViperEkura 648e4e177b feat(khaosz/trainer): 添加SchedulerCallback功能 2025-09-29 13:18:44 +08:00
ViperEkura 5163d3a47a fix(callback): 解决循环导入问题 2025-09-29 13:08:41 +08:00
ViperEkura b2f3fefa1b feat(callback): 为 TrainerCallback 及其子类添加文档字符串和未使用参数占位符 2025-09-29 12:48:01 +08:00
ViperEkura e52803ddc3 refactor(trainer): 将回调类移至独立文件并优化训练器结构 2025-09-29 12:00:25 +08:00
ViperEkura 8206c7855e fix(transformer): 调整注意力掩码处理逻辑 2025-09-29 11:31:42 +08:00
ViperEkura 816bc78894 feat(trainer): 引入训练器回调机制并重构训练流程 2025-09-29 11:31:31 +08:00
ViperEkura 92999fa9f6 fix(trainer): 修复训练器中配置引用错误的问题 2025-09-28 22:20:25 +08:00
ViperEkura 1c9063fd3d refactor(trainer): 统一参数命名以提升可读性 2025-09-28 22:14:24 +08:00
ViperEkura fa43ed2943 feat(trainer): 重构训练配置与策略工厂引入 2025-09-28 21:39:48 +08:00
ViperEkura 1169cfad82 fix(trainer): 修复多轮对话中的因果注意力掩码计算逻辑等 2025-09-28 15:15:19 +08:00
ViperEkura 25ec56a1f5 fix(trainer): 修复训练器恢复检查点时的学习率初始化问题 2025-09-28 14:38:02 +08:00
ViperEkura f25a249291 feat(khaosz): 优化模型参数保存与加载逻辑 2025-09-28 14:00:21 +08:00
ViperEkura 4fcdc87c95 feat(trainer): 重构数据集与策略模块以支持字典形式的数据返回 2025-09-27 14:11:27 +08:00
ViperEkura 9fbc9481b5 refactor(core): 修改注意力掩码处理函数并重命名参数 2025-09-27 13:37:10 +08:00
ViperEkura 053f4a4dad feat( StrategyFactory): 添加 SFT 策略初始化参数并完善工厂方法调用 2025-09-27 13:24:16 +08:00
ViperEkura 676fdd59d7 feat(strategy): 重构mask构建逻辑并优化策略工厂参数传递 2025-09-27 13:12:57 +08:00
ViperEkura a4443765ee Initial commit 2025-09-27 12:02:22 +08:00