MMEngine

功能简介

MMEngine是一个基于PyTorch框架用于训练深度学习模型的第三方库。MMEngine是OpenMMLab 2.0的基础库，用于基础runner搭建，兼容当前流行的模型库（例如Torchvision、Detectron2）并可以用更小的代码量完成训练任务。更多说明可前往官方文档。

安装方法

请参考MMCV完成MMCV的安装，再进行MMengine的安装。

安装原生MMEngine即可在昇腾NPU上使用。安装步骤如下：

1. 安装MMEngine。

   pip install mmengine>=0.5.0

2. 验证安装是否成功。

   python -c 'from mmengine.utils.dl_utils import collect_env;print(collect_env())'

   输出以下回显则证明安装成功。

   

使用方法

用户可以参考官网上的使用样例来体验MMengine的使用方法，并使用自动迁移将其代码迁移到NPU上运行。这个体验样例仅用80行代码，基于CIFAR-10数据集进行Resnet50模型的训练。