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