AstrAI/khaosz/trainer
ViperEkura d21682f97a fix(trainer): 修复检查点回调参数顺序和权重保存选项 2026-01-05 17:08:09 +08:00
..
__init__.py feat(trainer): 支持分布式训练配置与检查点加载优化 2025-12-19 19:34:39 +08:00
checkpoint.py fix(trainer): 修复检查点回调参数顺序和权重保存选项 2026-01-05 17:08:09 +08:00
metric_util.py feat(khaosz/trainer): 新增梯度统计工具函数并重构训练回调机制 2025-10-07 13:03:32 +08:00
schedule.py refactor(trainer): 优化trainer 结构 2025-12-07 21:23:05 +08:00
strategy.py feat(trainer): 改进模型输入和损失计算中的数据类型精度 2025-12-08 14:10:08 +08:00
train_callback.py fix(trainer): 修复检查点回调参数顺序和权重保存选项 2026-01-05 17:08:09 +08:00
train_context.py refactor(paralell): 优化并行设备指定方法 2025-12-26 20:54:33 +08:00
trainer.py fix(trainer): 修复检查点回调参数顺序和权重保存选项 2026-01-05 17:08:09 +08:00