首页 > Web开发 > 详细

Network slimming

时间:2020-02-07 09:10:33      阅读:81      评论:0      收藏:0      [点我收藏+]

一、paper&&github

       https://link.zhihu.com/?target=https%3A//arxiv.org/abs/1708.06519

  https://github.com/mengrang/Slimming-pytorch

二、本文压缩方法

       1、channel的裁剪原则

             使用BN中的gamma作为标准,如果该值较小,即认为其不重要,则该channel即被删减。

       2、损失函数

             网络中gamma值较小的可能不多,因此在训练的时候将其作为一个惩罚项,来加入损失函数。

      其中g取的是smooth_l1 norm。

             技术分享图片

 

 

        3、训练过程

              训练模型->采集模型(采集70%)->微调模型,这个过程可以多次进行。

 

              技术分享图片

 

 

 

 

 

参考 https://zhuanlan.zhihu.com/p/39761855 

Network slimming

原文:https://www.cnblogs.com/573177885qq/p/12271384.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!