Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)
MIT License
907 stars 214 forks source link

问题咨询:剪枝后通道数为0 #87

Open zhangtao1234567 opened 1 year ago

zhangtao1234567 commented 1 year ago

作者您好,非常感谢您的工作。在运行代码的时候,我对训练的模型进行剪枝的时候,出现剪枝后通道数为0的情况,torch和torchvision我用的是1.10.1和0.11.2,和说明文档中的不一样。我想咨询下,必须和文档中环境一样吗?是否可以通过调整参数改变这种问题呢?

zhangtao1234567 commented 1 year ago

按照文档的脚本运行方式解决了剪枝后通道为0的问题,还有就是模型微调后会变大,正常吗?vgg原始模型是160M左右,剪枝后是9M左右,然后微调恢复精度是18M左右

Uriccc commented 1 year ago

你好,请问是怎么解决的呢

zhangtao1234567 commented 1 year ago

你好,请问是怎么解决的呢

就是按照作者github上面的脚本一步一步运行,然后就没出现剪枝通道为0的情况了

Uriccc commented 1 year ago

你好,请问是怎么解决的呢

就是按照作者github上面的脚本一步一步运行,然后就没出现剪枝通道为0的情况了

你好,请问是怎么解决的呢

就是按照作者github上面的脚本一步一步运行,然后就没出现剪枝通道为0的情况了

谢谢,发现是我的地址有问题。已经没有这个情况了

cppcm123456 commented 1 year ago

你好,请问是怎么解决这个问题的,是那个地址有问题

Uriccc commented 1 year ago

是我torchload的时候用的相对地址如何出错的好像,我也记不清了

cppcm123456 commented 1 year ago

好的,写了