Pruning problem of deep grouped convolution models.

[zbzb-dlut]

Hello author! May I ask if this model can handle deep grouped convolution models? Additionally, for reparameterized models, is this method applicable during the training stage?

[aponte411]

hi @zbzb-dlut thanks for the question. Our response is late due to recent business travel.

Currently, for grouped conv, we only support the case where groups = out_channels, e.g., the depthwisw conv. However, adding support for different scenarios should be doable, which can be achieved by further development on

only_train_once/only_train_once/operation/operator.py

Line 153 in c5ec214

class Conv2dOTO(Operator):

As for the reparameterization, it depends on what type of reparameterization, in general, the answer is yes.