MemTorch icon indicating copy to clipboard operation
MemTorch copied to clipboard

Add Full Support for the `groups` Argument for Convolutional Layers

Open coreylammie opened this issue 4 years ago • 0 comments

Currently, all convolutional layer types support groups=1 and groups=in_channels operation modes. Support should be added for groups=2, where the operation becomes equivalent to having two conv layers side by side, each seeing half the input channels and producing half the output channels, and both subsequently concatenated.

coreylammie avatar Oct 07 '21 07:10 coreylammie