![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
模型剪枝
tang-0203
这个作者很懒,什么都没留下…
展开
-
ipython下运行代码出现<module '__main__' (built-in)> is a built-in class错误
这个问题是我在做模型剪枝的时候遇到的先载入VGG16模型,更改全连接层,使输出单元个数与自己待处理的数据类别个数一致。冻住卷积层,只训练全连接层。学习率设为0.0001,momentum=0.9。当设定的epoch数目满足的时候,把模型保存,使用命令:torch.save(model, "./cifar-10/5epochs_cifar10_vgg_model")4 . 然后在ipytho原创 2017-11-22 21:53:51 · 4791 阅读 · 0 评论 -
《Learning both Weights and Connections for Efficient Neural Networks》阅读笔记
参考链接:https://xmfbit.github.io/2018/03/14/paper-network-prune-hansong/主要思想:作者的方法分为三个步骤:Train Connectivity: 按照正常方法训练初始模型。作者认为该模型中权重的大小表征了其重要程度Prune Connection: 将初始模型中那些低于某个阈值的的权重参数置成0(即所谓剪枝)Re-Tra...转载 2019-04-16 22:59:57 · 546 阅读 · 0 评论