mindspore/model_zoo
i-robot a4ee2145a6 !23005 fix mobilenetv2 bugs on GPU
Merge pull request !23005 from zhaoting/master
2021-09-08 01:20:02 +00:00
..
community add directory in model_zoo 2020-07-14 11:44:19 +08:00
how_to_contribute add guide for modelscaffolding, update some readme 2021-08-28 16:23:48 +08:00
official !23005 fix mobilenetv2 bugs on GPU 2021-09-08 01:20:02 +00:00
research !22847 add a network EDSR 2021-09-07 13:47:50 +00:00
utils !22445 Add guide about modelscaffolding 2021-08-30 12:27:36 +00:00
OWNERS update modelzoo owners 2021-06-30 16:39:06 +08:00
README.md Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 2021-08-23 01:46:38 +08:00
README_CN.md Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 2021-08-23 01:46:38 +08:00
__init__.py Implements of masked seq2seq pre-training for language generation. 2020-06-20 15:48:49 +08:00

README.md

MindSpore Logo

Welcome to the Model Zoo for MindSpore

In order to facilitate developers to enjoy the benefits of MindSpore framework, we will continue to add typical networks and some of the related pre-trained models. If you have needs for the model zoo, you can file an issue on gitee or MindSpore, We will consider it in time.

  • SOTA models using the latest MindSpore APIs

  • The best benefits from MindSpore

  • Officially maintained and supported

Table of Contents

Announcements

Date News
September 25, 2020 Support MindSpore v1.0.0
September 01, 2020 Support MindSpore v0.7.0-beta
July 31, 2020 Support MindSpore v0.6.0-beta

Here is the ModelZoo for MindSpore which support different devices including Ascend, GPU, CPU and mobile.

If you are looking for exclusive models only for Ascend using different ML platform, you could refer to Ascend ModelZoo and corresponding gitee repository

Disclaimers

Mindspore only provides scripts that downloads and preprocesses public datasets. We do not own these datasets and are not responsible for their quality or maintenance. Please make sure you have permission to use the dataset under the datasets license. The models trained on these dataset are for non-commercial research and educational purpose only.

To dataset owners: we will remove or update all public content upon request if you dont want your dataset included on Mindspore, or wish to update it in any way. Please contact us through a Github/Gitee issue. Your understanding and contribution to this community is greatly appreciated.

MindSpore is Apache 2.0 licensed. Please see the LICENSE file.

License

Apache License 2.0

FAQ

  • Q: How to resolve the lack of memory while using PYNATIVE_MODE with errors such as Failed to alloc memory pool memory?

    A: PYNATIVE_MODE usually requires more memory than GRAPH_MODE, especially in training process which have to deal with back propagation. You could try using smaller batch size.