飞桨框架寒武纪 MLU 版支持模型¶

目前 Paddle MLU 版基于寒武纪 MLU370 系列板卡支持以下模型的单机单卡/单机多卡的训练。

图像分类¶

模型	领域	模型链接	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
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 下载即可获取所需的模型文件：