machine-learning - 无法从图层“conv1/7x7_s2/bn”复制参数 0 权重；形状不匹配。咖啡错误

Question

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

无法从图层“conv1/7x7_s2/bn”复制参数 0 权重；形状不匹配 h. 源参数形状为 1 64 1 1 (64)；目标参数形状为 64 (64)。要从 sc ratch 中学习该层的参数，而不是从保存的网络中复制，请重命名该层。

我检查了 deploy.prototxt 和 train_val.prototxt。尺寸是正确的。

有什么建议吗？

Answer 1

尝试编辑你的'deploy.prototxt'for 层'conv1/7x7_s2/bn'。添加

param {
  share_mode: PERMISSIVE
}

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

有关caffe.proto更多信息，请参阅。