mnist模型測試

參考部落格

1.準備資料

cd caffe
sudo ./data/mnist/get_mnist.sh #下載資料集
sudo ./examples/mnist/create_mnist.sh #轉換格式

2.訓練

sudo ./examples/mnist/train_lenet.sh

出現錯誤：

[email protected]-Lenovo-YOGA-710-14IKB:~/caffe$ sudo ./examples/mnist/train_lenet.sh
I1220 10:55:37.676761 20412 caffe.cpp:204
 
] Using GPUs 0
F1220 10:55:37.676882 20412 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode.
*** Check failure stack trace: ***
    @     0x7fe4c4c685cd  google::LogMessage::Fail()
    @     0x7fe4c4c6a433  google::LogMessage::SendToLog()
    @     0x7fe4c4c6815b  google::LogMessage::Flush()
    @     0x7fe4c4c6ae1e
 
  google::LogMessageFatal::~LogMessageFatal()
    @     0x7fe4c51a8a60  caffe::Caffe::SetDevice()
    @           0x40a78b  train()
    @           0x407208  main
    @     0x7fe4c344a830  __libc_start_main
    @           0x407ad9  _start
    @              (nil)  (unknown)
Aborted (core dumped)

分析：因為我沒有配置GPU，所以需要修改這個檔案。

• 嘗試解決：
  修改了檔案lenet_solver.prototxt:

# solver mode: CPU or GPU
solver_mode: CPU 

發現可以訓練了，這不簡單的嘛哈哈哈，不知道訓練得多長時間啊。壞了，忘記修改一下迭代次數了，這只是練習而已啊，表示一下就行了。。。
成功了：
竟然是經過了一萬次迭代，有點多了，剛才改成1000次還快一些。