Network List

View Source On Gitee

Model Zoo

Domain

Sub Domain

Network

Ascend

GPU

CPU

Computer Vision (CV)

Image Classification

AlexNet

Supported

Supported

Doing

Computer Vision (CV)

Image Classification

GoogleNet

Supported

Doing

Doing

Computer Vision (CV)

Image Classification

LeNet

Supported

Supported

Supported

Computer Vision (CV)

Image Classification

ResNet-50

Supported

Supported

Doing

Computer Vision (CV)

Image Classification

ResNet-101

Supported

Doing

Doing

Computer Vision (CV)

Image Classification

ResNext50

Supported

Supported

Doing

Computer Vision (CV)

Image Classification

VGG16

Supported

Doing

Doing

Computer Vision (CV)

Mobile Image Classification
Image Classification
Semantic Tegmentation

MobileNetV2

Supported

Supported

Doing

Computer Vision (CV)

Mobile Image Classification
Image Classification
Semantic Tegmentation

MobileNetV3

Doing

Supported

Doing

Computer Vision (CV)

Targets Detection

SSD

Supported

Doing

Doing

Computer Vision (CV)

Targets Detection

YoloV3-ResNet18

Supported

Doing

Doing

Computer Vision (CV)

Targets Detection

FasterRCNN

Supported

Doing

Doing

Computer Vision (CV)

Semantic Segmentation

DeeplabV3

Supported

Doing

Doing

Computer Vision(CV)

Targets Detection

WarpCTC

Doing

Supported

Doing

Natural Language Processing (NLP)

Natural Language Understanding

BERT

Supported

Doing

Doing

Natural Language Processing (NLP)

Natural Language Understanding

Transformer

Supported

Doing

Doing

Natural Language Processing (NLP)

Natural Language Understanding

SentimentNet

Doing

Supported

Supported

Natural Language Processing (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

Graph Neural Networks(GNN)

Text Classification

GCN

Supported

Doing

Doing

Graph Neural Networks(GNN)

Text Classification

GAT

Supported

Doing

Doing

Pre-trained Models

*It refers to the released MindSpore version. The hardware platforms that support model training are CPU, GPU and Ascend. As shown in the table below, ✓ indicates that the pre-trained model run on the selected platform.

Domain

Sub Domain

Network

Dataset

CPU

GPU

Ascend

0.5.0-beta*

Computer Vision (CV)

Image Classification

AlexNet

CIFAR-10

Download

Computer Vision (CV)

Image Classification

LeNet

MNIST

Download

Computer Vision (CV)

Image Classification

VGG16

CIFAR-10

Download

Computer Vision (CV)

Image Classification

ResNet-50

CIFAR-10

Download

Computer Vision (CV)

Targets Detection

YoloV3-DarkNet53

COCO 2014

Download

Natural Language Processing (NLP)

Natural Language Understanding

BERT_Base

zhwiki

Download

Natural Language Processing (NLP)

Natural Language Understanding

BERT_NEZHA

zhwiki

Download

Natural Language Processing (NLP)

Natural Language Understanding

Transformer

WMT English-German

Download