Eric-mingjie / network-slimming

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

the questions in resprune.py #62

Open kingoo123 opened 3 years ago

kingoo123 commented 3 years ago

作者您好: 我遇到了一个问题想向你请教,我用了你所给的方式训练了ResNet164,--s 0.00001,--dataset cifar100,然后在resprune中对训练好的网络进行剪枝,可是出现了其中某一层通道全部剪掉的情况,导致remaining channel=0,所以cfg中会出现0,以至于程序无法运行下去 image image

Eric-mingjie commented 3 years ago

应该需要多跑几次,当时我们也遇到过这个问题

kingoo123 commented 3 years ago

感谢mingjie的回复,我重新换了数据集就没有这个问题了,但是我还是很奇怪,如果是按照逻辑去推的话,这种情况是不应该发生的。

Eric-mingjie commented 3 years ago

有可能吧,如果一个层的scale factor都很小,就会被全部prune

sunanlin13174 commented 3 years ago

我刚刚也遇到了这个问题,现在明白了,感谢你们的讨论

fsj0122 commented 3 years ago

我刚刚也遇到了这个问题,现在明白了,感谢你们的讨论

你好,请问一下,ResNet164这个model 在哪里下载。