模型剪枝實操

性能優越的深度學習模型一般都有很大的參數量以及冗餘的參數量,這致使模型很難部署,相反,生物神經網絡都是用的是有效的稀疏鏈接,按照參數的重要性來減小壓縮參數,能夠有效地下降參數的存儲量、消耗的計算量以及硬件的電量。本文主要是教你們如何使用torch中的prune工具將參數稀疏化(torch.nn.utils.prune)python 需求環境 torch>1.4.0版本才具備該功能 須要加載的第三方
相關文章
相關標籤/搜索