Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

about resnet20 slimming

Danee-wawawa opened this issue · comments

commented

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?

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