老师可不可以给个模型剪枝，压缩的例子？

来源：9-5 签名函数，SavedModel和Keras模型到具体函数转换

xiabai84

2019-07-26

在深度学习网络层数过多的时候会造成模型过大或者影响性能，我看paper说是有方法可以减掉一些不重要的神经元，不知道老师能不能给几个例子或者Paper推荐？
按照这个剪枝方法我们是不是可以先用移除迁移学习最后2，3层，再加上自己的数据训练剩下的网络层，最后再剪枝减少参数和模型大小？

写回答

1回答

正十七

2019-07-31

已采纳

这个其实在tensorflow2.0中会有剪枝的case，只不过现在的测试版还没实现，等到正式版了加上这个功能了会再补充哈。

另外，论文的话可以搜下模型蒸馏，也是压缩的好办法。

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

回答 2

回答 2

回答 1

回答 1

回答 3