jacobgil/pytorch-pruning

ResNet model pruning

eeric opened this issue · 2 comments

eeric commented

Could it was pruned to ResNet model?
If ResNet model loaded, it occured error to FilterPrunner.

jiayouba120035 commented

hello, I have some problem with jacob's code, and want to ask you did you implement the code, when I run "python finetune --train",the test accuracy is always about 50%, but the train accuracy is more than 95%, did you have the problem like this, I see you want to prune ResNet, so asking you whether you implement the code. thank you in advance!

eeric commented

ResNet18 model pruned successfully,code in
https://github.com/eeric/channel_prune

  • 👍2