batchnorm-pruning icon indicating copy to clipboard operation
batchnorm-pruning copied to clipboard

Channel number mismatch

Open luluvi opened this issue 5 years ago • 0 comments

I replaced the resnet with the VGG and still had the same problem. In the sixth step, finetune's new model went wrong.The number of channels after pruning does not match the number of channels before pruning, is it because the pruning channel is not saved, and the original channel does not match?How do you solve this problem? Looking forward to your reply.Thank you.

RuntimeError: Given groups=1, weight of size [209, 128, 3, 3], expected input[64, 115, 12, 12] to have 128 channels, but got 115 channels instead

luluvi avatar Mar 31 '19 11:03 luluvi