昆仑 XPU 支持模型¶
飞桨框架在昆仑芯 XPU 上经验证的模型的支持情况如下:
训练支持¶
可进行单机单卡/单机多卡训练的模型,如下所示:
| 模型 | 领域 | 编程范式 | 可用的 CPU 类型 | 单机单卡支持 | 单机多卡支持 |
|---|---|---|---|---|---|
| ResNet50 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| ResNet101 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| MobileNet_v3 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| MobileNetV2 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| VGG19 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| VGG16 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| PP-LCNet | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| PP-HGNet | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| InceptionV4 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
| UNet | 图像分割 | 动态图 | X86(Intel) | 支持 | - |
| deeplabv3 | 图像分割 | 动态图 | X86(Intel) | 支持 | - |
| HRNet | 图像分割 | 动态图 | X86(Intel) | 支持 | - |
| PP-LiteSeq | 图像分割 | 动态图 | X86(Intel) | 支持 | - |
| PP-humansegv2 | 图像分割 | 动态图 | X86(Intel) | 支持 | - |
| PP-mating | 图像分割 | 动态图 | X86(Intel) | 支持 | - |
| MaskRcnn | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| FasterRcnn | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| fairmot | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| Yolov3-DarkNet53 | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| SSD-ResNet34 | 目标检测 | 动态图 | X86(Intel) | 支持 | 支持 |
| Yolov3-mobileNetv1 | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| PPYoloE | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| deepsort | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| ssd-mv1 | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| ssd-vgg16 | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| PP-picoDet | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| PPYolov2 | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
| OCR-DB | 文字检测 | 动态图 | X86(Intel) | 支持 | - |
| OCR-crnn | 文字检测 | 动态图 | X86(Intel) | 支持 | - |
| PPOCR-v2 | 文字检测 | 动态图 | X86(Intel) | 支持 | - |
| PPOCR-v3 | 文字检测 | 动态图 | X86(Intel) | 支持 | - |
| Bert-Base | NLP | 静态图 | X86(Intel) | 支持 | 支持 |
| Transformer | NLP | 静态图 | X86(Intel) | 支持 | 支持 |
| GPT-2 | NLP | 动态图 | X86(Intel) | 支持 | - |
| ernie-base | NLP | 动态图 | X86(Intel) | 支持 | - |
| ernie 3.0 medium | NLP | 动态图 | X86(Intel) | 支持 | - |
| lstm | NLP | 动态图 | X86(Intel) | 支持 | - |
| seq2seq | NLP | 动态图 | X86(Intel) | 支持 | - |
| DeepFM | 推荐 | 动态图 | X86(Intel) | 支持 | - |
| Wide&Deep | 推荐 | 动态图 | X86(Intel) | 支持 | - |
| dlrm | 推荐 | 动态图 | X86(Intel) | 支持 | - |
| deepspeech2 | 语音识别 | 动态图 | X86(Intel) | 支持 | - |
| speedyspeech | 语音合成 | 动态图 | X86(Intel) | 支持 | - |
| dqn | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| ppo | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| ddpg | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| A2C | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| TD3 | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| SAC | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| MADDPG | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| CQL | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| ES | 强化学习 | 动态图 | X86(Intel) | 支持 | - |
| pp-tsm | 视频分类 | 动态图 | X86(Intel) | 支持 | - |
模型放置在飞桨模型套件中,作为 github.com/PaddlePaddle 下的独立 repo 存在,git clone 下载即可获取所需的模型文件:
| 领域 | 套件名称 | 分支/版本 |
|---|---|---|
| 图像分类 | PaddleClas | develop |
| 目标检测 | PaddleDetection | develop |
| 图像分割 | PaddleSeg | develop |
| NLP | PaddleNLP | develop |
| OCR | PaddleOCR | dygraph |
| 推荐 | PaddleREC | master |