昆仑芯 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 |