2016-12-14 23 views
2

在Caffe訓練時,每次迭代都有Train和Test net輸出。我知道這是損失。但是,這是我批次的平均損失還是全部損失?這對分類和迴歸都是一樣的嗎?Caffe列車/測試網輸出是什麼意思?

例如,如果我有一批100個訓練樣例,並且我在該迭代中的損失爲100,那麼這是否意味着每個例子的平均損失是1?

回答

2

列車損失是上次培訓批次的平均損失。這意味着如果您的微型批次中有100個培訓示例,並且您在該迭代中的損失爲100,那麼每個示例的平均損失等於100.

測試損失也是平均損失,但是在所有測試批次。您可以指定測試批量大小和測試迭代次數。咖啡將採取這樣的小批量的#iter,評估他們的損失,併爲您提供一個平均值。如果#test_iter x batch_size == testset_size,您將在整個測試集中獲得平均值。

+0

清除污垢!謝謝! – Cassie

+0

根據Evan Shelhamer的說法,迭代的e行中報告的損失在解算器中指定的'average_loss'參數上平均:https://groups.google.com/forum/#!topic/caffe-users/WEhQ92s9Vus – Alex