Cannot copy param 0 weights from layer 'conv1/7x7_s2/bn'; shape mismatch. Caffe Error

330 views Asked by At

I trained pretrained GoogleNet(caffe) on Digits. The training went well and testing in Digits UI show no problem whatsoever. But when I export model and try to use example.py that digits provided. This error came up.

Cannot copy param 0 weights from layer 'conv1/7x7_s2/bn'; shape mismatc h. Source param shape is 1 64 1 1 (64); target param shape is 64 (64). To learn this layer's parameters from sc ratch rather than copying from a saved net, rename the layer.

I have checked the deploy.prototxt and train_val.prototxt. The dimension is correct.

Any suggestion?

1

There are 1 answers

2
Shai On

Try editing your 'deploy.prototxt' for layer 'conv1/7x7_s2/bn'. Add

param {
  share_mode: PERMISSIVE
}

For each of the trainable parameters of the layer.

See caffe.proto for more information.