沒有測試的Caffe培訓

Question

我正在使用Caffe在已知的圖像數據庫上訓練AlexNet。 我正在進行基准測試，並希望排除測試階段。

這是solver.prototxt的solver.prototxt：

net: "models/bvlc_alexnet/train_val.prototxt"
test_iter: 1000
test_interval: 1000
base_lr: 0.01
lr_policy: "step"
gamma: 0.1
stepsize: 100000
display: 20
max_iter: 450000
momentum: 0.9
weight_decay: 0.0005
snapshot: 10000
snapshot_prefix: "models/bvlc_alexnet/caffe_alexnet_train"
solver_mode: GPU

雖然我從未找到詳細說明所有原型文本選項的權威文檔，但Caffe教程中的注釋表明這個“ test_interval ”表示我們測試經過訓練的網絡之后的迭代次數。

我想我可以把它設置為零來關閉測試。 不。

 F1124 14:42:54.691428 18772 solver.cpp:140] Check failed: param_.test_interval() > 0 (0 vs. 0) *** Check failure stack trace: *** 

所以我將test_interval設置為一百萬，但當然，Caffe在迭代零點測試網絡。

 I1124 14:59:12.787899 18905 solver.cpp:340] Iteration 0, Testing net (#0) I1124 14:59:15.698724 18905 solver.cpp:408] Test net output #0: accuracy = 0.003 

如何在訓練時關閉測試？

Answer 1

Caffe的文檔在細節方面有些不足。 我最后被告知的是這種違反直覺的解決方案：

在solver.prototxt中，取test_iter和test_interval

test_iter: 1000
test_interval: 1000

並簡單地省略它們。 如果您想在開始時阻止測試，可以在@shai建議時添加一行：

test_initialization: false

Answer 2

你也有一面旗幟。 加

test_initialization: false

到你的'solver.prototxt' ，你就完成了;）