-
Notifications
You must be signed in to change notification settings - Fork 16
Open
Description
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
Metadata
Metadata
Assignees
Labels
No labels