Slimming-pytorch icon indicating copy to clipboard operation
Slimming-pytorch copied to clipboard

About prunning channels

Open Ariel666 opened this issue 6 years ago • 3 comments

Have you completed this project? I haven't seen the pruning channels codes.... Do you plan to actually prune the channels from the network?

Ariel666 avatar Jan 17 '19 07:01 Ariel666

I don't think you have understood my codes.Did you see the real_prune.py and the new_resnet.py?Actually, I do the real pruning, but just haven't fine tuned.

mengrang avatar Jan 17 '19 07:01 mengrang

Sorry, I didn't see the saving model codes after pruning. Can you show me that? Besides, how can I use new_resnet? Did I use real_prune to prune channels and use new_resnet to construct a new network?

Ariel666 avatar Jan 17 '19 09:01 Ariel666

Sorry for replying you later, First, I run the train.py to prune while training,then use the real_prune.py to remove the unimportant channels, finnally I load the pruned net weights to new_resnet

mengrang avatar Jan 20 '19 15:01 mengrang