38 KiB
Release 0.7.0-beta
Major Features and Improvements
MindSpore Training and Inference Framework
Ascend 910
- New models
- TinyBert: a smaller and faster version of BERT using transformer distillation for natural language understanding on GLUE benchmark.
- SE-ResNet50: add Squeeze-and-Excitation blocks(SE-Blocks) to the resnet50 network to improve channel interdependencies for image classification on ImageNet 2012 dataset.
- Inception V3: the third version of Inception convolutional architectures for image classification on ImageNet 2012 dataset.
- Frontend and user interface
- Embedding operator high-level packaging to support segmented by field for Wide&Deep.
- Load multi-node checkpoint into single-process to support host-device hybrid inference.
- Support Concat/Tile/Strideslice distributed operators.
- Support cumulative gradient and batch training split.
- Support variable parameter input for Cell object.
- Parameter mixed calculation optimization for pynative mode.
- Deep Probabilistic Programming
- Support statistical distributions classes used to generate stochastic tensors.
- Support probabilistic inference algorithms.
- Support BNN layers used to construct BNN in Graph mode.
- Support interfaces for the transformation between BNN and DNN in Graph mode.
- Support uncertainty estimation to estimate epistemic uncertainty and aleatoric uncertainty.
- User interfaces change log
- Executor and performance optimization
- Minspore graph compilation process performance improved by 20%.
- Decoupling C++ and Python modules to achieve separate compilation of core modules.
- Data processing, augmentation, and save format
- Support automatic data augmentation
- Support GNN distributed cache in single node
- Support ConcatDataset using distributed sampler
Other Hardware Support
- GPU platform
- New model supported: VGG16, ResNet101, DeepFM.
- Support some distributed operators in ResNet50 and Wide&Deep.
- Support automatic parallel for Wide&Deep.
- Support function funcsi (such as switch-case).
- Support distributed training with parameter server.
- Support GPU operator profiling.
- Performance optimization of the distributed training with allreduce.
- Performance optimization of the mixed precision training.
- Performance optimization of the pynative mode.
- Performance optimization of the convolution operator, batch normalization operator.
- CPU platform
- Support MobileNetV2 Re-Training: Re-train the network with different class number.
MindSpore Lite
- Converter
- Support third party model, including TFLite/Caffe/ONNX.
- Add 93 TFLite op.
- Add 24 Caffe op.
- Add 62 ONNX op.
- Add 11 optimized passes, include fusion/const fold.
- Support aware-training and Post-training quantization.
- CPU
- Add 100+ops,support fp32, int8/uint8, FP16 ops
- Support fast convolution algorithms: Sliding Window, Img2col + Gemm, Strassen, Winograd
- Support assembly/neon instruction.
- Support CPU fp16 and sdot on ARM v8.2+.
- GPU
- Add 20+ ops for OpenCL.
- Support image2D/buffer format.
- Optimize online initialization time.
- add optimized convolution1X1/3X3/depthwise/convolution_transposed for OpenCL.
- Tool & example
- Add benchmark and TimeProfile tools.
- Add image classification and object detection Android Demo.
Bugfixes
- Models
- Python API
- Executor
- fix etsnet train error when UnsegmentSum's first input shape is (1,) (!4573)
- fix bug of result error in while control flow because of unsupporting for value reference (!4103)
- fix bug of the output tensor does not carry device data type (!3774)
- fix bug of avoiding multi attr value are eliminated in pynative mode (!4225)
- fix bug of AssignAdd unable to work normally in multi-cases (!5171)
- GPU platform
- Data processing and Pro
Contributors
Thanks goes to these wonderful people:
Adel, Alexey, andy, andy_wangrui, anthonyaje, anzhengqi, askmiao, avakh, baihuawei, bingyaweng, BowenK, buxue, caifubi, CaoJian, caozhou, Cathy, changzherui, chenfei, chengxianbin, chenhaozhe, chenjianping, chentingting, chenzomi, chenzupeng, chujinjin, cjh9368, Corleone, cristoval, danish, dengyutao, eric, Eric, ervinzhang, etone-chan, fangzehua, fary86, fuzhiye, gengdongjie, genglishuai, Giancarlo, gongdaguo, gukecai, guohongzilong, GuoMengHao, hangq, hanhaocheng, hanhuifeng2020, hanjun996, Harshvardhan, He, heleiwang, hesham, hexia, Hoai, hongxing, huangdongrun, huanghui, huangxinjing, islam_amin, Jesse, jianghui58, jiangzhiwen, jin-xiulang, jinyaohui, jjfeing, John, Jonathan, jonyguo, kai00, kingfo, kpy, kswang, laiyongqiang, leilei_snow, leopz, Li, liangzelang, lianliguang, lichen_101010, lichenever, lihongkang, lilei, limingqi107, ling, lingyunli63, linqingke, lirongzhen1, liubuyu, liuwenhao4, liuxiao78, liuxiao93, liuzhongkai, Lixia, lixian, liyong, lizhenyu, looop5, luoyang, lvchangquan, lvliang, lvwenyuan, lyvette, mahdi, Mahdi, mamba_ni, maning202007, Margaret_wangrui, mayang, meixiaowei, meng_chunyang, ms_yan, nhussain, panbingao, panfengfeng, panyifeng, Payne, Peilin, peixu_ren, pengyongrong, Pengyongrong, qianlong, qujianwei, root, shenwei41, shibeiji, simson, songhonglei413, Su, sunsuodong, suteng, tao_yunhao, TFbunny, tinazhang, tom__chen, tony_liu2, tronzhang, VectorSL, wandongdong, wangdongxu, wanghua, wangmin, wangshaocong, wangzhe, wanyiming, Wei, wenchunjiang, wilfChen, WilliamLian, wsc, wukesong, wuweikang, wuxuejian, wuyongkang, xiefangqi, xuanyue, Xun, xutianchun, xuyongfei, yanghaitao, yangjie159, YangLuo, yangruoqi713, yangyongjie, yangzhenzhang, yankai, yao_yf, yelihua, yeyunpeng, Yi, yoni, yoonlee666, yuchaojie, yujianfeng, yuximiao, zhangxuetong, zhaizhiqiang, Zhang, zhangxinfeng3, zhangxuetong, zhangyihui, zhangz0911gm, zhanke, zhanyuan, zhaodezan, zhaoting, zhaozhenlong, zhengjun10, zhongligeng, zhoufeng, zhousiyi, zhouyaqiang, zhouyuanshen, Zichun, Zirui, zjun, zongha, ZPaC, lijiaqi, liangchenghui, wangminggui
Contributions of any kind are welcome!
Release 0.6.0-beta
Major Features and Improvements
Ascend 910 Training and Inference Framework
- New models
- There are official, research and community under modelzoo.
- Official is maintained with the newest APIs by MindSpore team, MaskRCNN are added.
- Research is uploaded by researchers for official review, and APIs may not be updated in time.
- Community reprints the relevant links of partner research results.
- Hub added on the same level as modelzoo, synchronous storage of materials needed for official hub web pages which will be launched soon.
- Support pre-trained models, few lines of code can be used to download and load pre-trained models, supporting inference or transfer learning.
- There are official, research and community under modelzoo.
- Frontend and user interface
- Supports user side operator compilation and graph execution error rendering.
- Uniform definition dynamic learning rate behavior in optimizers.
- Support IndexSlice in sparse expression.
- Support use parent construct method during construct.
- Support asynchronous execution save checkpoint file.
- Support implicit type conversion in pynative mode.
- User interfaces change log
- Executor and performance optimization
- Decouple C++ and python, so make the architecture more extensible.
- Parameter Server for distributed deep learning supported.
- Serving:a flexible service deployment framework for deep learning models.
- Memory reuse is enhanced, and the batch size of Bert large model is increased from 96 to 160 on a single server.
- Data processing, augmentation, and save format
- Support MindRecord save operator after date processing
- Support automatic fusion operator, such as decode/resize/crop
- Support CSV dataset loading
Other Hardware Support
- GPU platform
- New model supported: ResNext50, WarpCTC and GoogLeNet.
- Support hyperparametric search and data enhanced automl on GPU.
- Support Resnet50 automatic parallel in GPU backend.
Bugfixes
- Models
- Python API
- Fix assign used in while loop(!2720)
- Revert optimize the graph output of all nop node.(!2857)
- Print tensor as numpy.(!2859)
- Support weight decay for sparse optimizer(!2668)
- Fix BatchToSpaceND(!2741)
- Fixing type check mistakes of InplaceAdd and Inplace Sub ops(!2744)
- Change order param only equal to group param(!2748)
- Executor
- The performance of graph whith control flow is optimized(!2931)
- Fix bug of wrong number of tuple layers(!3390)
- Fix cpu multi graph memory exception(!3631)
- Enable data sync when calling operator without defining a cell(!3081)
- Fix argmaxwith value error in pynative mode on GPU(!3082)
- Fix precision error with fp16 input on pynative mode(!3196)
- Data processing
Contributors
Thanks goes to these wonderful people:
Alexey Shevlyakov, avakh, baihuawei, BowenK, buxue, caifubi, caojian05, Cathy Wong, changzherui, chenfei, chengxianbin, chenhaozhe, chenjianping, chentingting, chenzomi, chujinjin, Danish Farid, dayschan, dengwentao, dinghao, etone-chan, fangzehua, fary86, geekun, Giancarlo Colmenares, gong chen, gukecai, guohongzilong, hangangqiang, heleiwang, hesham, He Wei, hexia, hongxing, huangdongrun, huanghui, islam_amin, Jamie Nisbet, Jesse Lee, jiangjinsheng, jiangzhiwen, jinyaohui, jjfeing, jojobugfree, Jonathan Yan, jonyguo, Junhan Hu, Kang, kingfo, kouzhenzhong, kpy, kswang, laiyongqiang, leopz, liangzelang, lichenever, lihongkang, Li Hongzhang, lilei, limingqi107, lirongzhen1, liubuyu, liuchongming74, liuwenhao4, liuxiao, Lixia Chen, liyanliu, liyong, lizhenyu, lvliang, Mahdi, Margaret_wangrui, meixiaowei, ms_yan, nhussain, ougongchang, panfengfeng, panyifeng, peilinwang, Peilin Wang, pkuliuliu, qianlong, rick_sanchez, shibeiji, Shida He, shijianning, simson, sunsuodong, suteng, Tinazhang, Tron Zhang, unknown, VectorSL, wandongdong, wangcong, wangdongxu, wangdongxu6, wanghua, wangnan39, Wei Luning, wenchunjiang, wenkai, wilfChen, WilliamLian, wukesong, Xian Weizhao, Xiaoda Zhang, xiefangqi, xulei2020, xunxue, xutianchun, Yang, yanghaitao, yanghaitao1, yanghaoran, yangjie, yangjie159, YangLuo, Yanjun Peng, yankai, yanzhenxiang2020, yao_yf, Yi Huaijie, yoonlee666, yuchaojie, yujianfeng, zhangzhongpeng, zhangdengcheng, Zhang Qinghua, zhangyinxia, zhangz0911gm, zhaojichen, zhaoting, zhaozhenlong, zhoufeng, zhouneng, zhousiyi, Zirui Wu, Ziyan, zjun, ZPaC, lihongzhang, wangdongxu
Contributions of any kind are welcome!
Release 0.5.0-beta
Major Features and Improvements
Ascend 910 Training and Inference Framework
- New models
- ResNext50: a simple, highly modularized network architecture using aggregated resdiual transformations for image classification on ImageNet 2012 dataset.
- MASS: a pre-training method for sequence to sequence based language generation tasks on Text Summarization and Conversational Response Generation using News Crawls 2007-2017 dataset, Gigaword corpus and Cornell movie dialog corpus.
- Transformer: a neural network architecture for language understanding on WMT 2014 English-German dataset.
- GCN:Graph Convolutional Networks for the task of classification of nodes in a graph on Cora and Citeseer datasets.
- GAT:an attention-based graph neural network for node classification on Cora and CiteSeer dataset.
- Frontend and user interface
- Support tensor value and assignment of mixed tensor index in graph mode.
- Support tensor comparison, len operator, constexpr syntax, value and assignment of tensor index in pynative mode.
- Support converting MindSpore IR to pb format for infer model.
- Support print operator to write data directly on the hard disk.
- Add the double recursive programming solution for very high speed parallel strategy search in automatic parallel.
- User interfaces change log
- Allow the learning rate of AdamWeightDecayDynamicLR and Lamb to be 0(!1826)
- Restricting the entire network input parameter is Tensor(!1967)
- Turn shape and dtype into attributes instead of interfaces(!1919)
- Delete multitypefungraph(!2116)
- Refactor the callback module in an encapsulated way, use _CallbackManager instead of _build_callbacks(!2236)
- Delete EmbeddingLookup(!2163)
- Checkpoint add model_type(!2517)
- Executor and performance optimization
- Heterogeneous execution on CPU and Ascend devices supported, and is verified in Wide&Deep model.
- Quantitative training of MobileNetV2, Lenet and Resnet50 on Ascend-910 are supported.
- Support new fusion architecture, which can do fusion optimization across graphs and kernels to improve execution speed.
- Data processing, augmentation, and save format
- Support data processing pipeline performance profiling.
- Support public dataset loading, such as CLUE and Coco.
- Support more text processing, such as more tokenizers and vocab data.
- Support MindRecord padded data.
Other Hardware Support
- GPU platform
- New model supported: Bert / Wide&Deep.
- Support setting max device memory.
- CPU platform
- New model supported: LSTM.
Bugfixes
- Models
- Bert, Move Bert from
example
tomodel_zoo
, optimize network for better performance. (!1902) - VGG16, Move VGG16 from
example
tomodel_zoo
, optimize network for better accuracy. (!2645) - Alexnet, modify parameter setting to improve accuracy (!1364)
- Wide&Deep, Move Wide&Deep from
example
tomodel_zoo
, optimize network for better performance. (!2221)
- Bert, Move Bert from
- Python API
- Executor
- GPU platform
- Fix bug of summary feature invalid(!2173)
- Data processing
Contributors
Thanks goes to these wonderful people:
Alexey Shevlyakov, avakh, baihuawei, BowenK, buxue, caifubi, caojian05, Cathy Wong, changzherui, chenfei, chengxianbin, chenhaozhe, chenjianping, chentingting, chenzomi, chujinjin, Danish Farid, dayschan, dengwentao, dinghao, etone-chan, fangzehua, fary86, geekun, Giancarlo Colmenares, gong chen, gukecai, guohongzilong, hangangqiang, heleiwang, hesham, He Wei, hexia, hongxing, huangdongrun, huanghui, islam_amin, Jamie Nisbet, Jesse Lee, jiangjinsheng, jiangzhiwen, jinyaohui, jjfeing, jojobugfree, Jonathan Yan, jonyguo, Junhan Hu, Kang, kingfo, kouzhenzhong, kpy, kswang, laiyongqiang, leopz, liangzelang, lichenever, lihongkang, Li Hongzhang, lilei, limingqi107, lirongzhen1, liubuyu, liuchongming74, liuwenhao4, liuxiao, Lixia Chen, liyanliu, liyong, lizhenyu, lvliang, Mahdi, Margaret_wangrui, meixiaowei, ms_yan, nhussain, ougongchang, panfengfeng, panyifeng, peilinwang, Peilin Wang, pkuliuliu, qianlong, rick_sanchez, shibeiji, Shida He, shijianning, simson, sunsuodong, suteng, Tinazhang, Tron Zhang, unknown, VectorSL, wandongdong, wangcong, wangdongxu, wangdongxu6, wanghua, wangnan39, Wei Luning, wenchunjiang, wenkai, wilfChen, WilliamLian, wukesong, Xian Weizhao, Xiaoda Zhang, xiefangqi, xulei2020, xunxue, xutianchun, Yang, yanghaitao, yanghaitao1, yanghaoran, yangjie, yangjie159, YangLuo, Yanjun Peng, yankai, yanzhenxiang2020, yao_yf, Yi Huaijie, yoonlee666, yuchaojie, yujianfeng, zhangzhongpeng, zhangdengcheng, Zhang Qinghua, zhangyinxia, zhangz0911gm, zhaojichen, zhaoting, zhaozhenlong, zhoufeng, zhouneng, zhousiyi, Zirui Wu, Ziyan, zjun, ZPaC, lihongzhang, wangdongxu
Contributions of any kind are welcome!
Release 0.3.1-alpha
Major Features and Improvements
Ascend 910 Training and Inference Framework
- Frontend and User Interface
- Independent model init interface.
- Data processing, augmentation, and save format
- Support sample padding for minddataset.
Bugfixes
- Python API
- Fix bugs in the lars optimizer(!1894)
- Data processing
- Fix accuracy problem of RandomCropDecodeResize (!2340)
Release 0.3.0-alpha
Major Features and Improvements
Ascend 910 Training and Inference Framework
- New models
- DeepFM: a factorization-machine based neural network for CTR prediction on Criteo dataset.
- DeepLabV3: significantly improves over our previous DeepLab versions without DenseCRF post-processing and attains comparable performance with other state-of-art models on the PASCAL VOC 2007 semantic image segmentation benchmark.
- Faster-RCNN: towards real-time object detection with region proposal networks on COCO 2017 dataset.
- SSD: a single stage object detection methods on COCO 2017 dataset.
- GoogLeNet: a deep convolutional neural network architecture codenamed Inception V1 for classification and detection on CIFAR-10 dataset.
- Wide&Deep: jointly trained wide linear models and deep neural networks for recommender systems on Criteo dataset.
- Frontend and User Interface
- Complete numpy advanced indexing method. Supports value and assignment through tensor index.
- Some optimizers support separating parameter groups. Different parameter groups can set different
learning_rate
andweight_decay
. - Support setting submodule's logging level independently, e.g. you can set logging level of module
A
to warning and set logging level of moduleB
to info. - Support weights to be compiled according to shape to solve the problem of large memory overhead.
- Add some operators implement and grammar support in pynative mode. To be consistent with graph mode.
- User interfaces change log
- Executor and Performance Optimization
- Support doing evaluation while in training process, so that the accuracy of training can be easily obtained.
- Enable second-order optimization for resnet50, which can achieve 75.9% accuracy in 45 epochs (Resnet50 @ImageNet).
- Optimize pynative implementation and improve it's execution performance.
- Optimize summary record implementation and improve its performance.
- Data processing, augmentation, and save format
- Support simple text processing, such as tokenizer/buildvocab/lookup.
- Support padding batch.
- Support split or concat dataset.
- Support MindDataset reading from file list.
Other Hardware Support
- GPU platform
- New models supported: MobileNetV2, MobileNetV3.
- Support mixed precision training.
- Support device memory swapping.
Bugfixes
- Python API
- Executor
- Fix dropout,topK and addn errors in PyNative mode (!1285, !1138, !1033).
- Fix memory leaks after execution in PyNatvie mode (!1201).
- Fix HCCL failure in some special scenes (!1204, !1252).
- Fix SSD network when Select failed, cann't find kernel info(!1449).
- Fix Topk operator selection strategy bug between aicore and aicpu(!1367).
- Fix input memory size of 'assign' op unequal in control sink mode when assigning a data from one child graph to another child graph(!802).
- Fix allreduce ir inconsistency(!989).
- GPU platform
- Data processing
Contributors
Thanks goes to these wonderful people:
Alexey Shevlyakov, Amir Lashkari, anthony, baihuawei, biffex, buxue, caifubi, candanzg, caojian05, Cathy Wong, changzherui, chenfei, chengxianbin, chenhaozhe, chenzomi, chujinjin, cristoval, dengwentao, eric, etone-chan, fary86, gaojing, gengdongjie, gongchen, guohongzilong, guozhijian, heleiwang, hesham, He Wei, Hoai Linh Tran, hongxing, huangdongrun, huanghui, Jamie Nisbet, Jesse Lee, jiangjinsheng, jiangzhiwen, jinyaohui, jjfeing, jonwe, jonyguo, Junhan Hu, Kang, kingfo, kswang, laiyongqiang, leopz, lichenever, lihongkang, limingqi107, liubuyu, liuliyan2, liuwenhao4, liuxiao, liuxiao, liyong, lizhenyu, lvliang, Margaret_wangrui, meixiaowei, ms_yan, Nat Sutyanyong, ougongchang, panfengfeng, panyifeng, Peilin Wang, peixu_ren, qianlong, rick_sanchez, seatea, sheng, shijianning, simson, sunsuodong, Tinazhang, VectorSL, wandongdong, wangcong, wanghua, wangnan39, Wei Luning, wenchunjiang, wilfChen, WilliamLian, wsc, wukesong, wuxuejian, Xiaoda Zhang, xiefangqi, xulei2020, Yang, yangjie159, yangruoqi713, yangyongjie, yangzhenzhang, Yanjun Peng, yanzhenxiang2020, yao_yf, Yi Huaijie, yoonlee666, yujianfeng, YuJianfeng, yvetteliu, zhangdengcheng, Zhang Qinghua, zhangz0911gm, zhaojichen, zhaoting, zhaozhenlong, zhoufeng, zhouneng, zhousiyi, zhouyuanshen, Zirui Wu, Ziyan, zjun, ZPaC, lihongzhang
Contributions of any kind are welcome!
Release 0.2.0-alpha
Major Features and Improvements
Ascend 910 Training and Inference Framework
-
New models
- MobileNetV2: Inverted Residuals and Linear Bottlenecks.
- ResNet101: Deep Residual Learning for Image Recognition.
-
Frontend and User Interface
- Support for all python comparison operators.
- Support for math operators **,//,%. Support for other python operators like and/or/not/is/is not/ in/ not in.
- Support for the gradients of function with variable arguments.
- Support for tensor indexing assignment for certain indexing type.
- Support for dynamic learning rate.
- User interfaces change log
- DepthwiseConv2dNative, DepthwiseConv2dNativeBackpropFilter, DepthwiseConv2dNativeBackpropInput(!424)
- ReLU6, ReLU6Grad(!224)
- GeneratorDataset(!183)
- VOCDataset(!477)
- MindDataset, PKSampler(!514)
- map(!506)
- Conv(!226)
- Adam(!253)
- _set_fusion_strategy_by_idx, _set_fusion_strategy_by_size(!189)
- CheckpointConfig(!122)
- Constant(!54)
-
Executor and Performance Optimization
- Support parallel execution of data prefetching and forward/backward computing.
- Support parallel execution of gradient aggregation and forward/backward computing in distributed training scenarios.
- Support operator fusion optimization.
- Optimize compilation process and improve the performance.
-
Data processing, augmentation, and save format
- Support multi-process of GeneratorDataset/PyFunc for high performance
- Support variable batchsize
- Support new Dataset operators, such as filter,skip,take,TextLineDataset
Other Hardware Support
- GPU platform
- Use dynamic memory pool by default on GPU.
- Support parallel execution of computation and communication.
- Support continuous address allocation by memory pool.
- CPU platform
- Support for windows 10 OS.
Bugfixes
- Models
- Fix mixed precision bug for VGG16 model (!629).
- Python API
- Executor
- GPU platform
- Data processing
Contributors
Thanks goes to these wonderful people:
Alexey_Shevlyakov, Cathy, Chong, Hoai, Jonathan, Junhan, JunhanHu, Peilin, SanjayChan, StrawNoBerry, VectorSL, Wei, WeibiaoYu, Xiaoda, Yanjun, YuJianfeng, ZPaC, Zhang, ZhangQinghua, ZiruiWu, amongo, anthonyaje, anzhengqi, biffex, caifubi, candanzg, caojian05, casgj, cathwong, ch-l, chang, changzherui, chenfei, chengang, chenhaozhe, chenjianping, chentingting, chenzomi, chujinjin, dengwentao, dinghao, fanglei, fary86, flywind, gaojing, geekun, gengdongjie, ghzl, gong, gongchen, gukecai, guohongzilong, guozhijian, gziyan, h.farahat, hesham, huangdongrun, huanghui, jiangzhiwen, jinyaohui, jjfeing, jojobugfree, jonathan_yan, jonyguo, jzw, kingfo, kisnwang, laiyongqiang, leonwanghui, lianliguang, lichen, lichenever, limingqi107, liubuyu, liuxiao, liyong, liyong126, lizhenyu, lupengcheng, lvliang, maoweiyong, ms_yan, mxm, ougongchang, panfengfeng, panyifeng, pengyanjun, penn, qianlong, seatea, simson, suteng, thlinh, vlne-v1, wangchengke, wanghua, wangnan39, wangqiuliang, wenchunjiang, wenkai, wukesong, xiefangqi, xulei, yanghaitao, yanghaoran, yangjie159, yangzhenzhang, yankai10, yanzhenxiang2020, yao_yf, yoonlee666, zhangbuxue, zhangz0911gm, zhangzheng, zhaojichen, zhaoting, zhaozhenlong, zhongligeng, zhoufeng, zhousiyi, zjun, zyli2020, yuhuijun, limingqi107, lizhenyu, chenweifeng.
Contributions of any kind are welcome!
Release 0.1.0-alpha
Main Features
Ascend 910 Training and Inference Framework
- Recommended OS: Ubuntu 16.04 (or later) or EulerOS 2.5 or EulerOS 2.8
- Python version: 3.7.5
- Preset models
- ResNet-50: residual structure-based convolutional neural network (CNN) for image classification, which is widely used.
- AlexNet: classic CNN for image classification, achieving historical results in ImageNet LSVRC-2012.
- LeNet: classic CNN for image classification, which was proposed by Yann LeCun.
- VGG16: classic CNN for image classification, which was proposed by Oxford Visual Geometry Group.
- YoloV3: real-time object detection network.
- NEZHA: BERT-based Chinese pre-training network produced by Huawei Noah's Ark Laboratory.
- Execution modes
- Graph mode: provides graph optimization methods such as memory overcommitment, IR fusion, and buffer fusion to achieve optimal execution performance.
- PyNative mode: single-step execution mode, facilitating process debugging.
- Debugging capability and methods
- Save CheckPoints and Summary data during training.
- Support asynchronous printing.
- Dump the computing data.
- Support profiling analysis of the execution process performance.
- Distributed execution
- Support AllReduce, AllGather, and BroadCast collective communication.
- AllReduce data parallel: Each device obtains different training data, which accelerates the overall training process.
- Collective communication-based layerwise parallel: Models are divided and allocated to different devices to solve the problem of insufficient memory for large model processing and improve the training speed.
- Automatic parallel mode: The better data and model parallel mode can be predicted based on the cost model. It is recommended that this mode be used on ResNet series networks.
- Automatic differentiation
- Implement automatic differentiation based on Source to Source.
- Support distributed scenarios and automatic insertion of reverse communication operators.
- Data processing, augmentation, and save format
- Load common datasets such as ImageNet, MNIST, CIFAR-10, and CIFAR-100.
- Support common data loading pipeline operations, such as shuffle, repeat, batch, map, and sampler.
- Provide basic operator libraries to cover common CV scenarios.
- Support users to customize Python data augmentation operators through the Pyfunc mechanism.
- Support the access of user-defined datasets through the GeneratorDataset mechanism.
- Provide the MindSpore data format, data aggregation and storage, random access example, data partition, efficient parallel read, user-defined index, and dataset search.
- Convert user datasets to the MindSpore data format.
- After data processing and augmentation, provide training applications in feed and graph modes.
- FP32/16 mixed precision computation, supporting automatic and manual configuration
- Provide common operators such as nn, math, and array, which can be customized.
Inference Deployment
- Deploy models in MindSpore format on the Ascend 310 platform for inference.
- Save models in ONNX format.
- Support saving models in LITE format and running models based on the lightweight inference framework.
- Recommended OS: Android 4.3 or later
- Supported network type: LeNet
- Provide the generalization operators generated by TVM and operators generated after specific networks are tuned.
Other Hardware Support
- GPU platform training
- Recommended OS: Ubuntu 16.04
- CUDA version: 9.2 or 10.1
- CuDNN version: 7.6 or later
- Python version: 3.7.5
- NCCL version: 2.4.8-1
- OpenMPI version: 3.1.5
- Supported models: AlexNet, LeNet, and LSTM
- Supported datasets: MNIST and CIFAR-10
- Support data parallel.
- CPU platform training
- Recommended OS: Ubuntu 16.04
- Python version: 3.7.5
- Supported model: LeNet
- Supported dataset: MNIST
- Provide only the stand-alone operation version.
Peripherals and Tools
- [MindSpore Official Website] (https://www.mindspore.cn/)
- [MindInsight Visualization Debugging and Optimization] (https://gitee.com/mindspore/mindinsight)
- [MindArmour Model Security Hardening Package] (https://gitee.com/mindspore/mindarmour)
- [GraphEngine Computational Graph Engine] (https://gitee.com/mindspore/graphengine)