mmengine

收录理由

MMEngine 是 OpenMMLab 生态中众多项目共用的训练底座，凡是接触过 MMDetection、MMSegmentation 或 MMPose 的开发者，迟早都会和它打交道。它本身是一套务实的 PyTorch 训练引擎：配置系统灵活，注册器能让你挂载自定义组件，钩子生命周期则提供了对训练循环的细粒度控制，评估指标、日志和分布式工具也都已经内置。相比手写训练脚本，它给自定义训练流程提供了更清晰的结构，同时又不至于像完整框架那样带来太多约束。代码风格干净，文档也相当完善，把它当作一份成熟训练引擎如何组织各模块的参考也很有价值。检查点管理、实验可复现性以及标准化评估都能直接使用，省去自己重写这些样板逻辑的功夫。