【发布时间】：2018-06-18 21:58:20
【问题描述】：
我在一个包含 44 个类的自定义数据集上微调 CaffeNet，并收到以下错误：
inner_product_layer.cpp:64] 检查失败：K_ == new_K（9216 与 12544）输入大小与内部不兼容产品参数。
任何人都可以帮助解决这个问题吗？
更新
仅在将图像传递给经过训练的模型时才会出现此问题。
【问题讨论】：
标签： neural-network caffe
【发布时间】：2018-06-18 21:58:20
【问题描述】：
如果您更改预测的数量 (num_output)，您的权重 blob 的大小也会发生变化，并且您无法再从保存的 caffemodel 中对其进行初始化。您需要更改层的名称，然后 caffe 将使用随机权重初始化它（根据filler），您可以从那里进行微调
【讨论】：
-
我确实更改了图层的名称。其实训练很顺利。当我将图像传递给经过训练的模型时，就会出现问题。