[Caffe]:關於Check failed: shape[i] >= 0 (-2 vs. 0)錯誤
錯誤發現及分析
當使用caffe訓練網路的時候,出現如題錯誤,我們可以分析下日誌,日誌如下:
我們可以看出,在錯誤發生前生成的Top shape:64 2048 4 4
。也就是feature size是4 × 4的。然後再看看pool5對應的pooling layer的設定:
發現pooling的核大小被設定成7,這就是問題的所在了。
解決方案
- 改變輸入影象的大小,可以使feature map size變大
- 更改kernel_size的大小
第二種方法比較直接。