2016-12-06 123 views
0

我正在訓練一個簡單的Caffe神經網絡,來自MNIST的例子。我需要同時訓練兩件事情:在設置每600次迭代 如何在使用Caffe訓練時獲得訓練錯誤?

  • 在測試集結尾處的分類錯誤訓練

    1. 分類的錯誤,之後30000次迭代

    這可能嗎?

    我發現如何做1)通過更換測試集的訓練集和使用test_iter,但它似乎真的黑客,然後我不能有2)。例如,是否有一種簡單的方法來告訴Caffe計算每個迭代的訓練錯誤?

  • +0

    您看到的訓練錯誤是minibatches上的平均值,而不是整個集合。 – Shai

    +0

    是的,但是我需要在每次600次迭代後在整個數據集上發生錯誤。 –

    回答

    1
    1. 也許你可以在你的.prototxt文件中添加一個測試階段,並使用 Accuracy layer輸出的精度。
    2. 使用test_interval: 600設置測試間隔。
    +0

    我增加了第二階段,現在我在訓練期間擁有訓練準確性,但問題不在於我每次600次迭代都有測試和訓練準確性。但似乎這是我能做的最好的事情。 –

    +0

    這可能是最好的,可以在多個測試階段的咖啡因此我選擇了這個答案。 –