无法从层 'conv1/7x7_s2/bn' 复制参数 0 权重;形状不匹配。咖啡错误



我在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。尺寸正确。

有什么建议吗?

尝试编辑图层'conv1/7x7_s2/bn''deploy.prototxt'。加

param {
share_mode: PERMISSIVE
}

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

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

最新更新