jacobgil/pytorch-pruning

pruning conv layers with 'groups' > 1

iariav opened this issue · 0 comments

Hi,
first of all thanks for the awesome work.

i'm trying ti utilize your work to a more complicated model (xception). in this model there are conv layers with groups parameter which is different then 1 (specifically, it is equal to in_channels and output_channels). if i try to prune 1 filter out of such layer, i get an error since groups needs to be divisible by both in_channels and out_channels.

any idea how to overcome this issue?

currently as a workaround i'm skipping layers with groups > 1 but would really like to find a better solution,

thanks!