前记
例子基于docker -cpu 版本 caffe ,主要本人笔记本比较烂。而且希望快速开始caffe 的学习,不想折腾gpu 版本的nvidia-docker . 实际生产环境还是建议docker 的 gpu 版本。兼顾docker的简单及训练性能。
开始
- docker 中定位到目录:/opt/caffe
- 下载mnist数据集到这个目录下
./data/mnist/get_mnist.sh
- 转换格式,在examples/mnist生成了两个目录:mnist_test_lmdb和mnist_train_lmdb
./examples/mnist/create_mnist.sh # 运行命令时当前目录必须为/opt/caffe
- 在 /opt/caffe文件夹下运行, 开始训练
./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt
引用
https://www.jianshu.com/p/17841a4b755f