Linux下Caffe初学测试笔记

来源:互联网 发布:canon mp230 清零软件 编辑:程序博客网 时间:2024/06/05 07:52

感谢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 )等待结果

成功!



  
0 0