前記
例子基於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