模型压缩之通道剪枝channel pruning master.zip
对训练好的模型进行通道剪枝(channel pruning),分为两步进行:第一步是channel selection,采用LASSO regression来做,其实就是添加了一个L1范数来约束权重,因为L1范数可以使得权重更加稀疏,这样就可以把那些稀疏的channel剪掉;第二步是reconstruction,基于linear least squares来约束剪枝后输出的feature map要尽可能和减枝前的输出feature map相等,也就是最小二乘值越小越好。
文件列表
channel-pruning-master.zip
(预估有个25文件)
channel-pruning-master
calflop.sh
126B
logs
.gitkeep
0B
train.py
6KB
.github
issue_template.md
372B
combine.sh
108B
.gitmodules
79B
lib
暂无评论