ViperEkura e371908b54 fix : 保存 checkpoint 时 unwrap DDP/FSDP 避免 module. 前缀 ...

- 移除 state_dict_fn 参数
- _save_checkpoint 中先 unwrap_model 再 state_dict()

2026-05-28 18:10:04 +08:00

..

__init__.py

feat: 新增 Muon 优化器

2026-05-17 16:44:03 +08:00

metric_util.py

feat: 训练中新增验证循环

2026-05-17 16:12:42 +08:00

optim.py

perf: Muon step 改用 torch._foreach_* 批处理并移除 NS 迭代的冗余 bf16 转换

2026-05-23 19:50:12 +08:00

schedule.py

refactor : 移除 -> None 返回值标注，拆分 FSDP 参数，新增 mmap 数据集存储

2026-05-28 13:57:06 +08:00

strategy.py

refactor : 移除 -> None 返回值标注，拆分 FSDP 参数，新增 mmap 数据集存储

2026-05-28 13:57:06 +08:00

train_callback.py

fix : 保存 checkpoint 时 unwrap DDP/FSDP 避免 module. 前缀

2026-05-28 18:10:04 +08:00

train_context.py

refactor: 检查点加载重构，路径替代对象传递

2026-05-27 20:15:29 +08:00

trainer.py

refactor: 检查点加载重构，路径替代对象传递

2026-05-27 20:15:29 +08:00