飞桨对昆仑2代芯片的支持¶
飞桨自2.3rc版本起支持在昆仑2代芯片上(R200,R300)运行,经验证的模型训练的支持情况如下:
训练支持¶
可进行单机单卡/单机多卡训练的模型,如下所示:
模型 | 领域 | 编程范式 | 可用的CPU类型 | 单机单卡支持 | 单机多卡支持 |
---|---|---|---|---|---|
ResNet50 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
MobileNet_v3 | 图像分类 | 动态图 | X86(Intel) | 支持 | - |
UNet | 图像分割 | 动态图 | X86(Intel) | 支持 | - |
Yolov3-DarkNet53 | 目标检测 | 动态图 | X86(Intel) | 支持 | - |
SSD-ResNet34 | 目标检测 | 动态图 | X86(Intel) | 支持 | 支持 |
OCR-DB | 文字检测 | 动态图 | X86(Intel) | 支持 | - |
Bert-Base | NLP | 静态图 | X86(Intel) | 支持 | 支持 |
Transformer | NLP | 静态图 | X86(Intel) | 支持 | 支持 |
GPT-2 | NLP | 动态图 | X86(Intel) | 支持 | - |
DeepFM | 推荐 | 动态图 | X86(Intel) | 支持 | - |
Wide&Deep | 推荐 | 动态图 | X86(Intel) | 支持 | - |
模型放置在飞桨模型套件中,作为github.com/PaddlePaddle下的独立repo存在,git clone下载即可获取所需的模型文件:
领域 | 套件名称 | 分支/版本 |
---|---|---|
图像分类 | PaddleClas | develop |
目标检测 | PaddleDetection | develop |
图像分割 | PaddleSeg | develop |
NLP | PaddleNLP | develop |
OCR | PaddleOCR | dygraph |
推荐 | PaddleREC | master |
注:支持基于Kermel Primitive算子的昆仑2代芯片支持,点击这里。