AstrAI/astrai/parallel
ViperEkura 8cbf3f36e2 feat: 新增训练后端工厂框架
- BaseTrainingBackend 定义 prepare/accumulate/unwrap_model 抽象
- DDPTrainingBackend 支持全部 DDP 参数并通过 BackendFactory 注册
- unwrap_model 改为实例方法,由子类各自实现
2026-05-24 15:15:14 +08:00
..
__init__.py feat: 新增训练后端工厂框架 2026-05-24 15:15:14 +08:00
backend.py feat: 新增训练后端工厂框架 2026-05-24 15:15:14 +08:00
module.py refactor: 优化参数传递,清理导入样式 2026-04-03 22:06:32 +08:00
setup.py fix: 修复无法创建子进程的问题 2026-05-18 09:40:32 +08:00