飞桨框架 ROCm 版支持模型¶

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

图像分类¶

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

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

模型	领域	模型链接	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
ANN	图像分割	模型链接	动态图	支持	不支持	支持
BiSeNetV2	图像分割	模型链接	动态图	支持	不支持	支持
DANet	图像分割	模型链接	动态图	支持	不支持	支持
DeepLabV3+	图像分割	模型链接	动态图	支持	不支持	支持
Fast-SCNN	图像分割	模型链接	动态图	支持	不支持	支持
FCN	图像分割	模型链接	动态图	支持	不支持	支持
GCNet	图像分割	模型链接	动态图	支持	不支持	支持
GSCNN	图像分割	模型链接	动态图	支持	不支持	支持
HarDNet	图像分割	模型链接	动态图	支持	不支持	支持
OCRNet	图像分割	模型链接	动态图	支持	不支持	支持
U-Net	图像分割	模型链接	动态图	支持	不支持	支持
DecoupledSegNet	图像分割	模型链接	动态图	支持	不支持	支持
EMANet	图像分割	模型链接	动态图	支持	不支持	支持
ISANet	图像分割	模型链接	动态图	支持	不支持	支持
DNLNet	图像分割	模型链接	动态图	支持	不支持	支持

模型	领域	模型链接	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
BERT	NLP	模型链接	动态图	支持	支持	支持
XLNet	NLP	模型链接	动态图	支持	支持	支持
ELECTRA	NLP	模型链接	动态图	支持	支持	支持
Transformer	NLP	模型链接	动态图	支持	支持	支持
Seq2Seq	NLP	模型链接	动态图	支持	支持	支持
TextCNN	NLP	模型链接	动态图	支持	支持	支持
Bi-LSTM	NLP	模型链接	动态图	支持	支持	支持
LAC	NLP	模型链接	动态图	支持	支持	支持

模型	领域	模型链接	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
OCR-DB	文本检测	模型链接	动态图	支持	支持	支持
CRNN-CTC	文本识别	模型链接	动态图	支持	支持	支持
OCR-Clas	角度分类	模型链接	动态图	支持	支持	支持
OCR-E2E	字符识别	模型链接	动态图	支持	支持	支持

模型	领域	模型链接	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
SlowFast	视频分类	模型链接	动态图	支持	支持	支持
TSM	视频分类	模型链接	动态图	支持	支持	支持
TSN	视频分类	模型链接	动态图	支持	支持	支持
Attention-LSTM	视频分类	模型链接	动态图	支持	支持	支持
BMN	视频分类	模型链接	动态图	支持	支持	支持
StNet	视频分类	模型链接	动态图	支持	支持	支持

模型	领域	模型链接	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
Transformer TTS	语音合成	模型链接	动态图	支持	支持	支持
WaveFlow	语音合成	模型链接	动态图	支持	支持	支持

模型	领域	模型链接	编程范式	训练单机多卡支持	训练多机多卡支持	推理支持
Pix2Pix	风格迁移	模型链接	动态图	支持	支持	支持
CycleGAN	风格迁移	模型链接	动态图	支持	支持	支持
StyleGAN V2	人脸生成	模型链接	动态图	支持	支持	支持
Wav2Lip	唇形合成	模型链接	动态图	支持	支持	支持
ESRGAN	图像超分	模型链接	动态图	支持	支持	支持
EDVR	视频超分	模型链接	动态图	支持	支持	支持

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