Iterative Pruning

模型剪枝原理 文献通过修剪网络中不重要的权值,减小网络参数,主要分三个步骤: (1)正常训练整个网络. (2)修剪不重要的连接,一般认为权值比较大的比较重要,因此设定一个阈值,剪去小于阈值的权值. (3)重新训练网络,采用稀疏矩阵保存网络参数. 源码分析: 以MNIST CNN 模型为例,下载github源码: git clone https://github.com/garion9013/imp
相关文章
相关标签/搜索