1.MindSpore 安装
默认已安装anaconda和python(具体方法网络教程很多),创建一个python 3.7.5的新环境,在cmd命令下,进入新创建的环境,然后直接安装MindSpore 框架,本次安装是cpu+windows64+pip ,使用命令:
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.2.1/MindSpore/cpu/windows_x64/mindspore-1.2.1-cp37-cp37m-win_amd64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
具体教程见https://www.mindspore.cn/install
2.下载项目
lenet5项目详见:https://gitee.com/mindspore/course/tree/master/lenet5
MNIST数据集:http://yann.lecun.com/exdb/mnist/, ,下载数据集后,解压,创建一个文件夹MNIST,并分别将train和test数据集放在该文件下,文件目录结构为
lenet5
├── MNIST
├── test
├── t10k-images-idx3-ubyte
└── t10k-labels-idx1-ubyte
└── train
├── train-images-idx3-ubyte
└── train-labels-idx1-ubyte
└── main.py
下载了数据集后需要解压,不然程序会报错。
3.main.py函数解读
环境配置
准备数据集
构建神经网络
训练模型
主函数,确定数据接口
4.训练过程
由于代金券还没到账,首先试了下在cpu上训练,只要在环境配置中令
device_target='CPU'
然后,直接运行程序就ok啦~可以改改num_epochs试试呀
def train(data_dir, lr=0.01, momentum=0.9, num_epochs=3):
5.写在最后
作为是一名人工智能刚入坑的娃,可能标注会有错误,欢迎大家指出更正。MindSpore框架还有很多不熟悉,以后会进一步学习,进步!