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,否则可能出现错误