search

Eric-mingjie / network-slimming

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

about resnet20 slimming #66

Open Danee-wawawa opened 3 years ago

Danee-wawawa commented 3 years ago

Hi, thank you for your work. My model is resnet20. If I want to use your slimming algorithm, just modify this parameter --depth to 20? Is there anything else that needs to be modified?

Eric-mingjie commented 3 years ago

You need to add the resnet20 architecture and also add the corresponding channel selection layer.