网络支持

Model Zoo

领域 子领域 网络 Ascend GPU CPU
计算机视觉(CV) 图像分类(Image Classification) AlexNet Supported Supported Doing
计算机视觉(CV) 图像分类(Image Classification) GoogleNet Supported Doing Doing
计算机视觉(CV) 图像分类(Image Classification) LeNet Supported Supported Supported
计算机视觉(CV) 图像分类(Image Classification) ResNet-50 Supported Supported Doing
计算机视觉(CV) 图像分类(Image Classification) ResNet-101 Supported Doing Doing
计算机视觉(CV) 图像分类(Image Classification) ResNext50 Supported Supported Doing
计算机视觉(CV) 图像分类(Image Classification) VGG16 Supported Doing Doing
计算机视觉(CV) 移动端图像分类(Mobile Image Classification)
目标检测(Image Classification)
语义分割(Semantic Tegmentation)
MobileNetV2 Supported Supported Doing
计算机视觉(CV) 移动端图像分类(Mobile Image Classification)
目标检测(Image Classification)
语义分割(Semantic Tegmentation)
MobileNetV3 Doing Supported Doing
计算机视觉(CV) 图像分类(Image Classification) ResNet-50 Supported Supported Doing
计算机视觉(CV) 目标检测(Targets Detection) SSD Supported Doing Doing
计算机视觉(CV) 目标检测(Targets Detection) YoloV3-ResNet18 Supported Doing Doing
计算机视觉(CV) 目标检测(Targets Detection) FasterRCNN Supported Doing Doing
计算机视觉(CV) 语义分割(Semantic Segmentation) DeeplabV3 Supported Doing Doing
计算机视觉(CV) 目标检测(Targets Detection) WarpCTC Doing Supported Doing
自然语言处理(NLP) 自然语言理解(Natural Language Understanding) BERT Supported Doing Doing
自然语言处理(NLP) 自然语言理解(Natural Language Understanding) Transformer Supported Doing Doing
自然语言处理(NLP) 自然语言理解(Natural Language Understanding) SentimentNet Doing Supported Supported
自然语言处理(NLP) 自然语言理解(Natural Language Understanding) MASS Supported Doing Doing
推荐(Recommender) 推荐系统、点击率预估(Recommender System, CTR prediction) DeepFM Supported Doing Doing
推荐(Recommender) 推荐系统、搜索、排序(Recommender System, Search ranking) Wide&Deep Supported Doing Doing
图神经网络(GNN) 文本分类(Text Classification) GCN Supported Doing Doing
图神经网络(GNN) 文本分类(Text Classification) GAT Supported Doing Doing

预训练模型

*代表MindSpore已发布的版本号,支持网络训练的硬件平台有CPU、GPU和Ascend,以下表格中 ✓ 代表模型是基于选中的硬件平台训练而来。

领域 子领域 网络 CPU GPU Ascend 0.5.0-beta*
计算机视觉(CV) 图像分类(Image Classification) AlexNet 下载
计算机视觉(CV) 图像分类(Image Classification) LeNet 下载
计算机视觉(CV) 图像分类(Image Classification) VGG16 下载
计算机视觉(CV) 图像分类(Image Classification) ResNet-50 下载
计算机视觉(CV) 目标检测(Targets Detection) YoloV3-DarkNet53 下载
自然语言处理(NLP) 自然语言理解(Natural Language Understanding) BERT_Base 下载
自然语言处理(NLP) 自然语言理解(Natural Language Understanding) BERT_NEZHA 下载
自然语言处理(NLP) 自然语言理解(Natural Language Understanding) Transformer 下载