剪枝基础与实战(5): 剪枝代码详解

对模型进行剪枝，我们只对有参数的层进行剪枝，我们基于BatchNorm2d对通道重要度 $\gamma$ 参数进行稀释训练。对BatchNorm2d及它的前后层也需要进行剪枝。主要针对有参数的层：Conv2d、BatchNorm2d、Linear。但是我们不会对Pool2d 层进行剪枝，因为Pool2d只用来做下采样，没有可学习的参数，不用处理

在这里插入图片描述

0 VGG 模型代码

在这里插入图片描述

import<

相关阅读:
Python函数进阶：探索高级函数特性与技巧
Java 七牛云OSS 上传文件，删除图片，刷新CDN
【概念认知】关于“体系”概念的个人认识
图计算 on nLive：Nebula 的图计算实践
深度学习与CV教程(10) | 轻量化CNN架构 (SqueezeNet,ShuffleNet,MobileNet等)
Java并发编程解析 | 解析AQS基础同步器的设计与实现
基于RFID技术的烟草供应链数字化管理方案
Rancher清理节点
设计模式之【门面模式（外观模式）】
Archive Team: The Twitter Stream Grab

原文地址：https://blog.csdn.net/weixin_38346042/article/details/132524857