Caffe：inner_product_layer.cpp：64]检查失败：K_ == new_K

Question

我在一个有44个类的自定义数据集上调整CaffeNet并收到以下错误：
inner_product_layer.cpp：64]检查失败：K_ == new_K（9216 vs. 12544）输入尺寸与内部产品参数不兼容。
有人可以协助解决这个问题吗？

的更新
只有在将图像传递给训练模型时才会出现问题。

Answer 1

如果您更改预测数量（num_output），您的体重blob的大小也会发生变化，您无法再从已保存的caffemodel中启动它。您需要更改图层的名称，然后caffe将使用随机权重（根据filler）初始化它，您可以从那里进行微调