迁移准备

模型选取

在选取模型时,尽可能选取权威PyTorch模型实现仓作为标杆,包括但不限于PyTorch(imagenet/vision等)、facebookresearch(Detectron/detectron2等)和open-mmlab(mmdetection/mmpose等)。

若用户没有模型迁移相关的经验,建议先基于PyTorch官网提供的Imagenet数据集训练脚本main.py,根据模型迁移中的迁移流程,理解一下迁移的要点和方法。

模型移植评估

模型是否可以迁移成功主要取决于模型算子是否支持昇腾AI处理器,故需要对模型算子对昇腾AI处理器的支持性进行评估。