可以支持自定义网络结构剪枝吗？比如不是典型的网络结构
GeneralJing opened this issue · comments
可以的，参考 https://github.com/Sanster/pytorch-network-slimming/blob/master/gen_schema.py 脚本，修改下里面的 build_model 部分
好的，我研究一下，多谢
Sanster / pytorch-network-slimming
A package to make do Network Slimming a little easier