Iterative Pruning

模型剪枝原理 文獻通過修剪網絡中不重要的權值,減小網絡參數,主要分三個步驟: (1)正常訓練整個網絡. (2)修剪不重要的連接,一般認爲權值比較大的比較重要,因此設定一個閾值,剪去小於閾值的權值. (3)重新訓練網絡,採用稀疏矩陣保存網絡參數. 源碼分析: 以MNIST CNN 模型爲例,下載github源碼: git clone https://github.com/garion9013/imp
相關文章
相關標籤/搜索