AstrAI/khaosz/trainer
ViperEkura fd7ee2895a refactor(paralell): 优化并行设备指定方法 2025-12-26 20:54:33 +08:00
..
__init__.py feat(trainer): 支持分布式训练配置与检查点加载优化 2025-12-19 19:34:39 +08:00
checkpoint.py feat(trainer): 支持分布式训练配置与检查点加载优化 2025-12-19 19:34:39 +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 feat(trainer): 支持分布式训练配置与检查点加载优化 2025-12-19 19:34:39 +08:00
train_context.py refactor(paralell): 优化并行设备指定方法 2025-12-26 20:54:33 +08:00
trainer.py refactor(paralell): 优化并行设备指定方法 2025-12-26 20:54:33 +08:00