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