基于pytorch实现模型剪枝
创始人
2024-05-26 18:09:05
  • 一,剪枝分类
  • 二,PyTorch 的剪枝
  • 三,总结
  • 参考资料

一,剪枝分类

所谓模型剪枝,其实是一种从神经网络中移除"不必要"权重或偏差(weigths/bias)的模型压缩技术。关于什么参数才是“不必要的”,这是一个目前依然在研究的领域。

1.1,非结构化剪枝

非结构化剪枝(Unstructured Puning)是指修剪参数的单个元素,比如全连接层中的单个权重、卷积层中的单个卷积核参数元素或者自定义层中的浮点数(scaling floats)。其重点在于,剪枝权重对象是随机的,没有特定结构,因此被称为非结构化剪枝

1.2,结构化剪枝

与非结构化剪枝相反,结构化剪枝会剪枝整个参数结构。比如,丢弃整行或整列的权重,或者在卷积层中丢弃整个过滤器(Filter)。

1.3,本地与全局修剪

剪枝可以在每层(局部)或多层/所有层(全局)上进

相关内容

热门资讯

科源制药受诉讼拖累盈利降逾70... 长江商报消息 ●长江商报记者 沈右荣受子公司诉讼拖累,2月24日晚,科源制药(301281.SZ)下...
ETF投资“神器”一网打尽!华... (来源:合富永道)你还在为选哪家基金公司的ETF工具发愁吗?别急,今天咱们把市面上三家主打指数/ET...
广西北流镇通报龙安桥桥下火情:... 本文转自【北流融媒】;关于北流镇龙安桥桥梁底下发生火情的情况通报2026年2月25日14时许,位于北...
祝大家都一次过!四六级考试今日... 本文转自【央视新闻客户端】;2025年下半年全国大学英语四、六级考试(CET)成绩查询服务今日(2月...
守正创新强根基 凝心铸魂启新程 (来源:邯郸日报)转自:邯郸日报 裴晓丽 思想政治工作是党的优良传统和政治优势,更是县级电影公司凝心...