从 MMCV 到 MMEngine，架构升级，体验升级！

深度学习任务模块辣么多，各种涨点 trick 层出不穷，还要科学管理实验数据，太难了，我学不动。 o(╥﹏╥)o

为啥不试试 MMCV 嘞，各种 SOTA 模块开箱即用，还能自动保存实验配置~

嘶，真香！

MMCV 怎么在 IterBasedRunner 下使用 EMAHook 啊？

MMCV 怎么在模型里获取 Epoch 啊？

需求 +1000...

我修、我改、我加！啊啊啊啊啊啊啊啊啊啊啊奥利给！

MMCV 作为深度学习基础框架的先驱，一经开源就受到了社区同学的广泛关注。然而随着深度学习任务的需求变得越来越多样化，MMCV 早期的架构设计也开始难以满足日益复杂的需求。

因此 OpenMMLab 2.0 发布了新一代训练架构 MMEngine，以统一的执行引擎，灵活支持了 20 个以上的计算机视觉任务。

如果你刚入坑深度学习，还在苦恼于如何管理实验数据，搭建实验框架，不妨试一试 MMEngine，用最少的代码体验最丰富的功能。

如果你已经使用过 MMCV，我们强烈推荐你升级使用 MMEngine。我们广泛收集了 MMCV 开源以来，大家在使用过程中遇到的问题，并在 MMEngine 中一一解决优化。

啊，怕迁移太麻烦？不用怕，MMEngine 同样提供了详尽的迁移文档，能够让大家无痛迁移。

不少小伙伴们吐槽过 MMCV 的文档过于简单，介绍不全等，MMEngine 痛定思痛，提供了超级详细的教程文档和 API 文档，助力大家快速上手 MMEngine！欢迎大家阅读交流，给出宝贵的意见。

用户文档链接：https://mmengine.readthedocs.io/zh_CN/latest/

小至 typo，大至文档内容和结构，欢迎小伙伴们提出宝贵意见！如果文档中有代码示例跑不通，欢迎随时在 issue 中反馈。

更加强大的 Runner

• 功能更全面！Runner is all you need！配置 Runner 等于配置了整个训练流程
• 搭配更灵活！数据集和评测指标能够自由搭配，优化流程更加灵活
• 使用更简单！构造 Runner，调用 Runner 就能完成训练验证和测试

使用 MMEngine 的 Runner，我们可以用不到 80 行代码训练 CIFAR10。

具体可参考：https://mmengine.readthedocs.io/zh_CN/latest/get_started/15_minutes.html

我们既可以通过构建好的实例去初始化 Runner，也能通过配置文件初始化 Runner，乐高式的构建流程让你的深度学习任务更加灵活。

具体可参考：

https://mmengine.readthedocs.io/zh_CN/latest/tutorials/runner.html

更加丰富的 Hook 点位

MMEngine 归纳总结了 MMCV 开发过程中，社区同学、下游仓库提出的各种意见，重新设计了 Runner 的训练/验证/测试流程和 Hook 点位。

基于新的 Hook 设计，我们可以在训练、测试、验证前后，加载权重前后做一些自定义的操作，让 Hook 的功能更加丰富，此外 Hook 的各个点位也会接受更多参数，让我们能够在 Hook 中更加灵活的实现功能。

更加统一的模型

当你基于 MMCV 开发项目时：

• 是否苦恼于不知道应该实现模型的哪些接口？
• 是否疑惑于模型接口的标准是什么，怎么实现接口才符合设计规范？
• 是否疲于 ctrl c 、ctrl v 拷贝模型通用的功能代码？

赶快来体验 MMEngine 吧！BaseModel 提供了模型的通用接口，也约定了每个接口的规范，让自定义模型继承 BaseModel，并且按照接口规范去实现模型，相信会有更加流畅的开发体验！

如下图所示，模型的输入输出需要满足数据加载器、评测器、优化器封装的接口约定，因此 BaseModel 定义了 model 和各个模块之间的接口规范：

模型和各个模块之间具体的数据流详见用户文档：

https://mmengine.readthedocs.io/zh_CN/latest/tutorials/model.html

更加通用的优化器封装

如果你之前使用 MMCV 开发自己的项目，会发现要想开启混合精度训练，需要同时配置多个模块，例如给模型设置 fp16_enabled 、启用 Fp16OptimizerHook，还需要给模型的各个接口加上类似 auto_fp16 的装饰器，少写一处都会无法顺利开启混合精度训练。

如果你是 MMGeneration、MMEditing 的用户，并且对 MMDetection 或 MMClassification 等算法库有着一定了解，可能会抱怨，为啥我不能在 MMGeneration 中使用 MMCV 里实现的各种各样的 OptimizerHook 呢，还需要在模型代码里实现混合精度训练、梯度累加等逻辑。

MMEngine 的优化器封装（OptimWrapper ）解决了上述问题：

MMEngine 同样提供了 OptimWrapper 的详细图文攻略，包你无痛上手！

https://mmengine.readthedocs.io/zh_CN/latest/tutorials/optim_wrapper.html

更加灵活的参数调度器：scheduler

MMCV-1.x、PyTorch 里实现了的调度器，MMEngine 全都有（不仅支持学习率调整，也支持动量调整）。

不仅如此，MMEnine 实现的调度器还支持调度器之间的自由组合。

更加灵活的模型精度评测

在模型验证和模型测试中，通常需要对模型精度做定量评测。在 OpenMMLab 1.0 中，模型的精度验证是和 dataset 绑定的，这就对切换评测指标带来了困难。

因此，我们在 MMEngine 中实现了评测指标（Metric）和评测器（Evaluator）模块来完成模型评测，实现评测过程和数据集类的解耦。

• Metric：负责实现根据测试数据和模型预测结果，完成模型特定精度指标的计算。
• Evaluator：类则位于 Metric 的上层，为 Metric 提供格式转换以及分布式通信的支持。

如下图所示，同一个 Evaluator 可以包含多个 Metric，使用不同的评价指标来评估模型。

看到这你或许会疑惑，Runner、Hook 这些核心组件都在 MMEngine 里实现了，那 MMCV 还有啥嘞？架构升级后，我们还能用 MMCV 做哪些事情？

在回答这个问题之前，大家不妨思考一下，OpenMMLab 的算法库里，哪个模块的代码重复率很高？没有错！那就是数据变换（pipeline）。

MMCV 2.0 归纳总结了各个算法数据变换方向的需求，实现了一系列功能强大的数据变换。

除此之外，MMCV 也会有一些模块增删，下一期我们就来具体聊聊架构升级后的 MMCV，敬请期待哦~

想了解更多 MMEngine 相关内容，欢迎查看直播回放视频~