模型剪枝实操

性能优越的深度学习模型一般都有很大的参数量以及冗余的参数量,这致使模型很难部署,相反,生物神经网络都是用的是有效的稀疏链接,按照参数的重要性来减小压缩参数,能够有效地下降参数的存储量、消耗的计算量以及硬件的电量。本文主要是教你们如何使用torch中的prune工具将参数稀疏化(torch.nn.utils.prune)python 需求环境 torch>1.4.0版本才具备该功能 须要加载的第三方
相关文章
相关标签/搜索