search

Sanster / pytorch-network-slimming

A package to make do Network Slimming a little easier
47 stars 10 forks source link

可以支持自定义网络结构剪枝吗？比如不是典型的网络结构 #1

Closed GeneralJing closed 3 years ago

Sanster commented 3 years ago

可以的，参考 https://github.com/Sanster/pytorch-network-slimming/blob/master/gen_schema.py 脚本，修改下里面的 build_model 部分

GeneralJing commented 3 years ago

好的，我研究一下，多谢