caffe例子 - mnist

前记

例子基于docker -cpu 版本 caffe ,主要本人笔记本比较烂。而且希望快速开始caffe 的学习,不想折腾gpu 版本的nvidia-docker . 实际生产环境还是建议docker 的 gpu 版本。兼顾docker的简单及训练性能。

开始

  1. docker 中定位到目录:/opt/caffe
  2. 下载mnist数据集到这个目录下
./data/mnist/get_mnist.sh
  1. 转换格式,在examples/mnist生成了两个目录:mnist_test_lmdb和mnist_train_lmdb
 ./examples/mnist/create_mnist.sh   # 运行命令时当前目录必须为/opt/caffe
  1. 在 /opt/caffe文件夹下运行, 开始训练
./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

训练结果

引用

https://www.jianshu.com/p/17841a4b755f

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章