感谢http://www.cnblogs.com/denny402/p/5075490.html
初学caffe,陆续贴上自己的测试记录,以便查看,环境:ubuntu14.04+cuDNN2.0+caffev0.11 +Digits。
1 mnist 手写体测试
采用Lenet模型
1 )准备数据
转换为LMDB或者leveldb数据格式,mnist_train_lmdb和mnist_test_lmdb,里面存放的data.mdb和lock.mdb,这一步直接参考denny402的博客。
2) 配置文件
一个是lenet_solver_prototxt,一个是train_lenet.prototxt,看自己配置有没有GPU,默认是GPU,若只有CPU,要记得改solver.
要调整参数,都在这里面。
3 )运行
值得注意的是:必须要在caffe根目录下!!!如:/home/user/caffe
# sudo time sh examples/mnist/train_lenet.sh
4 )等待结果
成功!