剪枝部分：请问如果我的模型结构较为复杂，分支较多那么我如何修改剪枝的代码来遍历模型获取每个weight呢？

[zhishao]

毕竟不是所有模型都是像您代码里提供的nin一样这么简单的网络结构

[666DZY666]

确实，剪枝通用性有待加强。