mnist训练步骤

Caffe搭建与MNIST训练
本文详细介绍了在Windows环境下,使用CUDA7.5和cuDNN5搭建Caffe深度学习框架的过程,包括编译生成caffe.exe和libcaffe库,以及如何使用MNIST数据集进行LeNet模型的训练。注意,训练时避免使用cuDNN6,以免出现错误。

1.编译caffe,生成caffe.exe

2.编译生成libcaffe库

3.mnist训练

a.将制作好的数据集拷贝到example/mnist目录下,测试数据mnist_test_lmdb,训练数据 mnist_train_lmdb

b.修改lenet_solver.prototxt文件,如图:

c.修改lenet_train_test.prototxt文件,如图:

d.写测试脚本文件 train.bat,内容如下:

以上步骤完成后运行脚本开始训练。

附上一张训练结果图

本次训练环境为:Windows7, cuda7.5,cudnn5

注意不要使用cudnn6,否则可能出现错误

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值