Skip to content

寒武纪 MLU 支持模型

飞桨框架在寒武纪 MLU 上经验证的模型的支持情况如下:

图像分类

模型领域模型链接编程范式训练单机多卡支持训练多机多卡支持推理支持
ResNet50图像分类模型链接动态图支持支持支持
VGG16/19图像分类模型链接动态图支持支持支持
InceptionV4图像分类模型链接动态图支持支持支持
MobileNetV3图像分类模型链接动态图支持支持支持

目标检测

模型领域模型链接编程范式训练单机多卡支持训练多机多卡支持推理支持
YOLOv3目标检测模型链接动态图支持支持支持
PP-YOLO目标检测模型链接动态图支持支持支持
SSD目标检测模型链接动态图支持支持支持
Mask R-CNN目标检测模型链接动态图支持支持支持
Mask R-CNN + FPN目标检测模型链接动态图支持支持支持

图像分割

模型领域模型链接编程范式训练单机多卡支持训练多机多卡支持推理支持
DeepLabV3+图像分割模型链接动态图支持不支持支持
U-Net图像分割模型链接动态图支持不支持支持

自然语言处理

模型领域模型链接编程范式训练单机多卡支持训练多机多卡支持推理支持
BERTNLP模型链接动态图支持支持支持
TransformerNLP模型链接动态图支持支持支持
Bi-LSTMNLP模型链接动态图支持支持支持

字符识别

模型领域模型链接编程范式训练单机多卡支持训练多机多卡支持推理支持
OCR-DB文本检测模型链接动态图支持支持支持
CRNN-CTC文本识别模型链接动态图支持支持支持
OCR-Clas角度分类模型链接动态图支持支持支持
OCR-E2E字符识别模型链接动态图支持支持支持

模型套件

模型放置在飞桨模型套件中,各领域套件是 github.com/PaddlePaddle 下的独立 repo,git clone 下载即可获取所需的模型文件:

领域套件名称分支/版本
图像分类PaddleClasdevelop
目标检测PaddleDetectiondevelop
图像分割PaddleSegdevelop
自然语言处理PaddleNLPdevelop
字符识别PaddleOCRdevelop

基于 MIT 许可发布 共建 共享 共管