无法复制来自图层＆＃39; conv1 / 7x7_s2 / bn＆＃39;的param 0权重;形状不匹配。 Caffe错误

Question

我在Digits上训练了预训练的GoogleNet（caffe）。培训进展顺利，在Digits UI中测试显示没有任何问题。但是当我导出模型并尝试使用example.py提供的数字时。出现了这个错误。

  

无法从图层＆＃39; conv1 / 7x7_s2 / bn＆＃39;中复制参数0权重形状不对称   H。源param形状是1 64 1 1（64）;目标参数形状是64（64）。要从sc学习这一层的参数   棘手而不是从保存的网络中复制，重命名该图层。

我检查了deploy.prototxt和train_val.prototxt。维度是正确的。

有什么建议吗？

Answer 1

尝试修改图层'deploy.prototxt'的{​​{1}}。添加

'conv1/7x7_s2/bn'

对于图层的每个可训练参数。

有关详细信息，请参阅caffe.proto。