寒武纪 MLU 支持模型
飞桨框架在寒武纪 MLU 上经验证的模型的支持情况如下:
图像分类
| 模型 |
领域 |
模型链接 |
编程范式 |
训练单机多卡支持 |
训练多机多卡支持 |
推理支持 |
| ResNet50 |
图像分类 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| VGG16/19 |
图像分类 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| InceptionV4 |
图像分类 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| MobileNetV3 |
图像分类 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
目标检测
| 模型 |
领域 |
模型链接 |
编程范式 |
训练单机多卡支持 |
训练多机多卡支持 |
推理支持 |
| YOLOv3 |
目标检测 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| PP-YOLO |
目标检测 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| SSD |
目标检测 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| Mask R-CNN |
目标检测 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| Mask R-CNN + FPN |
目标检测 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
图像分割
| 模型 |
领域 |
模型链接 |
编程范式 |
训练单机多卡支持 |
训练多机多卡支持 |
推理支持 |
| DeepLabV3+ |
图像分割 |
模型链接 |
动态图 |
支持 |
不支持 |
支持 |
| U-Net |
图像分割 |
模型链接 |
动态图 |
支持 |
不支持 |
支持 |
自然语言处理
| 模型 |
领域 |
模型链接 |
编程范式 |
训练单机多卡支持 |
训练多机多卡支持 |
推理支持 |
| BERT |
NLP |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| Transformer |
NLP |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| Bi-LSTM |
NLP |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
字符识别
| 模型 |
领域 |
模型链接 |
编程范式 |
训练单机多卡支持 |
训练多机多卡支持 |
推理支持 |
| OCR-DB |
文本检测 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| CRNN-CTC |
文本识别 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| OCR-Clas |
角度分类 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
| OCR-E2E |
字符识别 |
模型链接 |
动态图 |
支持 |
支持 |
支持 |
模型套件
模型放置在飞桨模型套件中,各领域套件是 github.com/PaddlePaddle 下的独立 repo,git clone 下载即可获取所需的模型文件:
| 领域 |
套件名称 |
分支/版本 |
| 图像分类 |
PaddleClas |
develop |
| 目标检测 |
PaddleDetection |
develop |
| 图像分割 |
PaddleSeg |
develop |
| 自然语言处理 |
PaddleNLP |
develop |
| 字符识别 |
PaddleOCR |
develop |