来自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

回答

相关问题