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?
Try editing your
'deploy.prototxt'
for layer'conv1/7x7_s2/bn'
. AddFor each of the trainable parameters of the layer.
See
caffe.proto
for more information.