he-y / soft-filter-pruning

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks
https://arxiv.org/abs/1808.06866
374 stars 74 forks source link

如何处理残差结构中skip connection的裁剪? #35

Open coolKeen opened 1 year ago

coolKeen commented 1 year ago

@he-y 谢谢作者分享代码,看了下Mask类,好像没找到如何处理残差结构中skip connection,如下图: image 输入特征裁剪的通道和经过第二个conv之后的出来的特征裁剪的通道可能不一样,但它们需要相加,想请教下soft filter pruning是如何处理这种情况的?谢谢🙏