昆仑芯 XPU 支持模型

飞桨框架在昆仑芯 XPU 上通过精度验证的模型情况如下:

模型库 模型类型 模型名称 训练 推理
PaddleX 图像分类 ResNet50 √ √
PaddleX 图像分类 PP-LCNet_x1_0 √ √
PaddleX 图像分类 MobileNetV3_small_x1_0 √ √
PaddleX 图像分类 SwinTransformer_small_patch4_window7_224 √ √
PaddleX 图像分类 SwinTransformer_base_patch4_window7_224 √ √
PaddleX 图像分类 PP-HGNet_small √ √
PaddleX 图像分类 PP-HGNetV2-B0 √ √
PaddleX 图像分类 PP-HGNetV2-B4 √ √
PaddleX 图像分类 PP-HGNetV2-B6 √ √
PaddleX 图像分类 CLIP_vit_base_patch16_224 √ √
PaddleX 图像分类 CLIP_vit_large_patch14_224 √ √
PaddleX 图像分类 PP-ShiTuV2_rec_CLIP_vit_base √ √
PaddleX 图像分类 PP-ShiTuV2_rec_CLIP_vit_large √ √
PaddleX 目标检测 PP-YOLOE_plus-S √ √
PaddleX 目标检测 PP-YOLOE_plus-M √ √
PaddleX 目标检测 PP-YOLOE_plus-L √ √
PaddleX 目标检测 PP-YOLOE_plus-X √ √
PaddleX 目标检测 PicoDet-S √ √
PaddleX 目标检测 PicoDet-L √ √
PaddleX 目标检测 STFPM √ √
PaddleX 语义分割 PP-LiteSeg-T √ √
PaddleX 文本检测 PP-OCRv4_server_det √ √
PaddleX 文本检测 PP-OCRv4_mobile_det √ √
PaddleX 文本识别 PP-OCRv4_server_rec √ √
PaddleX 文本识别 PP-OCRv4_mobile_rec √ √
PaddleX 文本识别 ch_SVTRv2_rec √ √
PaddleX 版面分析 PicoDet_layout_1x √ √
PaddleX 公式识别 LaTeX_OCR_rec √ √
PaddleX 图像异常检测 STFPM √ √
PaddleX 时序预测 DLinear √ √
PaddleX 时序预测 RLinear √ √
PaddleX 时序预测 NLinear √ √
PaddleX 时序预测 PatchTST √ √
PaddleX 时序预测 TimesNet √ √
PaddleX 时序预测 TiDE √ √
PaddleX 时序异常检测 DLinear_ad √ √
PaddleX 时序异常检测 PatchTST_ad √ √
PaddleX 时序异常检测 TimesNet_ad √ √
PaddleX 时序异常检测 Nonstationary_ad √ √
PaddleX 时序异常检测 AutoEncoder_ad √ √
PaddleX 时序分类 TimesNet_cls √ √
PaddleNLP 自然语言理解模型 BERT √ √