batchnorm-pruning
batchnorm-pruning copied to clipboard
Channel number mismatch
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