YOLOv3-complete-pruning icon indicating copy to clipboard operation
YOLOv3-complete-pruning copied to clipboard

为什么shortcut_prune剪枝方法中,上采样前的卷积层不能进行剪枝?

Open worldzhang1996 opened this issue 4 years ago • 1 comments

在prune_utils.py文件中的 line 60-62 中代码如下: #上采样层前的卷积层不裁剪 ignore_idx.add(84) ignore_idx.add(96)

按照我的个人理解,上采样的话是只影响H和W两个维度,BN剪枝的话主要是对channel这个维度进行修剪。如果是对上采样前的卷积层进行剪枝,那么其实并不影响上采样操作?

worldzhang1996 avatar Nov 24 '20 06:11 worldzhang1996

你好,请问理解了吗,我也有相同的困惑,望解答一下谢谢

hzq-zjm avatar Apr 15 '21 06:04 hzq-zjm