您的位置:首页 > 运维架构 > Linux

Linux下Caffe初学测试笔记

2016-03-22 20:27 483 查看
感谢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 )等待结果

成功!

  
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  ubuntu 测试 Caffe