torchinfo icon indicating copy to clipboard operation
torchinfo copied to clipboard

Add the "groups" column to show number of channel groups.

Open andravin opened this issue 1 year ago • 0 comments

Is your feature request related to a problem? Please describe. torch.nn.Conv2d has a "groups" attribute set to a value greater than 1 for EfficientNet, ResNeXt, ConvNeXt, and other current vision models. Currently, torchinfo does not show group information for Conv2d layers.

Describe the solution you'd like Add a "groups" column to the supported information columns.

Describe alternatives you've considered It is possible to infer the number of groups from the output size, input size, and macs count, but this is too much work. Better to show it.

Additional context I will submit a pull request with the feature implemented.

andravin avatar Aug 22 '24 03:08 andravin