飞桨框架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下载即可获取所需的模型文件：