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

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