Mindspore r0.7版本运行Lenet_MNIST

Mindspore r0.7版本运行Lenet_MNIST

一、环境准备

  • 建议使用anaconda去创建一个单独容器去使用mindspore
    至于怎么去创建,可以参考前一张博客。

1) 安装mindspore r0.7
去官网上www.mindspore.cn寻找r0.7版本使用pip安装

pip install
https://ms-release.obs.cn-north-4.myhuaweicloud.com/0.7.0-beta/MindSpore/cpu/windows_x64/mindspore-0.7.0-cp37-cp37m-win_amd64.whl

2)拷贝代码
去码云mindspore上拷贝lenet代码
查看分支,获取链接
Mindspore r0.7版本运行Lenet_MNIST

二、运行准备

1)代码准备

根据以上,git代码

https://gitee.com/mindspore/mindspore.git

查看git之后代码,在你的当前文件路径下,不知道的可以是使用pwd命令

cd mindspore\model_zoo\official\cv\lenet

进入目录后将train.py和eval.py两文件修改一下

Mindspore r0.7版本运行Lenet_MNIST

改成CPU的
Mindspore r0.7版本运行Lenet_MNIST

2)数据准备
数据下载

http://yann.lecun.com/exdb/mnist/

Mindspore r0.7版本运行Lenet_MNIST
将这四个文件train文件归train,test归test文件,两个为一组。分别解压成Date/train和Data/test目录下,建议不要把压缩包放进去(可能会报错)

将Data这个文件放在lenet目录下

三、开始练丹

切换到lenet目录

cd mindspore\model_zoo\official\cv\lenet

执行训练代码

python train.py

就可以开始训练了,训练后会产生ckpt文件,放在了ckpt目录下Mindspore r0.7版本运行Lenet_MNIST

Mindspore r0.7版本运行Lenet_MNIST

执行验证代码

python eval.py --ckpt_path=./ckpt/checkpopint_lenet-10_1875.ckpt

Mindspore r0.7版本运行Lenet_MNIST

四、结束

本次是mindspore第四期运行作业。以上如有不足或者错误地方还望见谅和指正。一定积极学习改正。