寒武纪 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 |
