來自Caffe：inner_product_layer.cpp：64]檢查失敗：K_ == new_K

我微調CaffeNet上的自定義數據集44類，並收到以下錯誤：
inner_product_layer.cpp：64]檢查失敗：K_ == new_K（9216與12544）輸入大小與內部產品參數不兼容。
任何人都可以幫助解決這個問題嗎？

UPDATE
只有在將圖像傳遞給訓練好的模型時纔會出現此問題。來自Caffe：inner_product_layer.cpp：64]檢查失敗：K_ == new_K

來源

2017-08-14 Anjenson

如果您更改預測數量（num_output），您的權重blob的大小也會更改，並且您不能再從保存的caffemodel中初始化它。您需要更改圖層的名稱，然後caffe會用隨機權重初始化它（根據filler），您可以從那裏微調

來源

2017-08-14 15:15:11 Shai

我確實更改了圖層的名稱。其實訓練進展順利。將圖像傳遞給訓練好的模型時出現問題。 – Anjenson

來自Caffe：inner_product_layer.cpp：64]檢查失敗：K_ == new_K

回答

相關問題