RuntimeError: Given groups=1, weight of size [15, 14, 1, 1], expected input[64, 16, 32, 32] to have 14 channels, but got 16 channels instead

Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)

MIT License

907 stars 214 forks source link

RuntimeError: Given groups=1, weight of size [15, 14, 1, 1], expected input[64, 16, 32, 32] to have 14 channels, but got 16 channels instead #79

Open lovegit2021 opened 2 years ago

lovegit2021 commented 2 years ago

您好，我按照作者的流程跑resnet164的剪枝，在finetune的时候出现剪枝模型和网络结构加载不上的情况，如下报错：RuntimeError: Given groups=1, weight of size [15, 14, 1, 1], expected input[64, 16, 32, 32] to have 14 channels, but got 16 channels instead。请问是什么原因啊？

okkkkkkkkkkkkk commented 1 year ago

您好，请问您的这个问题解决了吗，我也遇到了类似的问题，可以给我点建议吗