Skip to content

昆仑 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-BaseNLP静态图X86(Intel)支持支持
TransformerNLP静态图X86(Intel)支持支持
GPT-2NLP动态图X86(Intel)支持-
ernie-baseNLP动态图X86(Intel)支持-
ernie 3.0 mediumNLP动态图X86(Intel)支持-
lstmNLP动态图X86(Intel)支持-
seq2seqNLP动态图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 下载即可获取所需的模型文件:

领域套件名称分支/版本
图像分类PaddleClasdevelop
目标检测PaddleDetectiondevelop
图像分割PaddleSegdevelop
NLPPaddleNLPdevelop
OCRPaddleOCRdygraph
推荐PaddleRECmaster

基于 MIT 许可发布 共建 共享 共管