飞桨框架ROCm版支持模型

来源:百度浏览 834 扫码分享 2021-06-06 23:17:01

飞桨框架ROCm版支持模型

飞桨框架ROCm版支持模型

目前Paddle ROCm版基于海光CPU(X86)和DCU支持以下模型的单机单卡/单机多卡的训练与推理。

图像分类

模型	领域	模型readme	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
ResNet50	图像分类	模型链接	动态图	支持	支持	支持
ResNet101	图像分类	模型链接	动态图	支持	支持	支持
SE_ResNet50_vd	图像分类	模型链接	动态图	支持	支持	支持
VGG16	图像分类	模型链接	动态图	支持	支持	支持
InceptionV4	图像分类	模型链接	动态图	支持	支持	支持
GoogleNet	图像分类	模型链接	动态图	支持	支持	支持
MobileNetV3	图像分类	模型链接	动态图	支持	支持	支持
AlexNet	图像分类	模型链接	动态图	支持	支持	支持
DenseNet121	图像分类	模型链接	动态图	支持	支持	支持

目标检测

模型	领域	模型readme	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
YOLOv3	目标检测	模型链接	动态图	支持	支持	支持
Faster R-CNN	目标检测	模型链接	动态图	支持	支持	支持
Faster R-CNN + FPN	目标检测	模型链接	动态图	支持	支持	支持
Mask R-CNN	目标检测	模型链接	动态图	支持	支持	支持
Mask R-CNN + FPN	目标检测	模型链接	动态图	支持	支持	支持
Cascade Faster R-CNN	目标检测	模型链接	动态图	支持	支持	支持
Cascade Mask R-CNN	目标检测	模型链接	动态图	支持	支持	支持
SSD	目标检测	模型链接	动态图	支持	支持	支持
BlazeFace	人脸检测	模型链接	动态图	支持	支持	支持
FaceBoxes	人脸检测	模型链接	静态图	支持	支持	支持
Libra R-CNN	目标检测	模型链接	静态图	支持	支持	支持
RetinaNet	目标检测	模型链接	静态图	支持	支持	支持

更多分类

模型	领域	模型readme	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
DeepLabV3	图像分割	模型链接	动态图	支持	不支持	支持
HRNet	图像分割	模型链接	动态图	支持	不支持	支持
BERT	NLP	模型链接	动态图	支持	不支持	支持
Transformer	NLP	模型链接	动态图	支持	不支持	支持
Seq2Seq	NLP	模型链接	动态图	支持	不支持	支持
Bi-LSTM	NLP	模型链接	动态图	支持	不支持	支持
DeepFM	推荐系统	模型链接	动态图	支持	不支持	支持
Wide&Deep	推荐系统	模型链接	动态图	支持	不支持	支持
Word2Vec	推荐系统	模型链接	动态图	支持	不支持	支持
NCF	推荐系统	模型链接	动态图	支持	不支持	支持
TSM	视频分类	模型链接	动态图	支持	支持	支持
StNet	视频分类	模型链接	动态图	支持	支持	支持
Attention-LSTM	视频分类	模型链接	动态图	支持	支持	支持
Transformer TTS	语音合成	模型链接	动态图	支持	不支持	支持
WaveFlow	语音合成	模型链接	动态图	支持	不支持	支持

模型套件

模型放置在飞桨模型套件中，各领域套件是 github.com/PaddlePaddle 下的独立repo，git clone下载即可获取所需的模型文件：

领域	套件名称	分支/版本
图像分类	PaddleClas	release/2.1
目标检测	PaddleDetection	release/2.0
图像分割	PaddleSeg	release/v2.0
NLP	PaddleNLP	develop
OCR	PaddleOCR	release/2.1
推荐系统	PaddleRec	master
视频分类	PaddleVideo	develop
语音合成	Parakeet	release/v0.2

当前内容版权归百度或其关联方所有，如需对内容或内容相关联开源项目进行关注与资助，请访问百度 .

本文档使用 BookStack 构建

展开/收起文章目录