环境
- python 3.9
- numpy 1.24.1
- pytorch 2.0.0+cu117
结构剪枝
-
类型
1.滤波器剪枝(Filter-level):对卷积核组进行纵向的修剪
2.通道剪枝(channel-level):对卷积核组进行横向的修剪
3.层剪枝(layer-level):直接删除整个卷积层
一、滤波器剪枝、通道剪枝
二 、剪枝代码
# -*- coding: utf-8 -*-import numpy as np#结构化剪枝#类型：# 1.滤波器剪枝(Filter-level):对卷积核组进行纵向的修剪# 2.通道剪枝(channel-level):对卷积核组进行横向的修剪# 3.层剪枝(layer-level):直接删除整个卷积层def prune_conv_layer(conv_layer, prune_method, percentile=20, vis = False): """ :param prune_method: 剪枝方式 :param percentile: 去掉低于百分比分位数的位置 :param
