nntoolbox.vision.components.inception module

class nntoolbox.vision.components.inception.FactoredConvolutionalLayer(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', activation=<class 'torch.nn.modules.activation.ReLU'>, normalization=<class 'torch.nn.modules.batchnorm.BatchNorm2d'>)[source]

Bases: torch.nn.modules.container.Sequential

Factor a k x k convolution into 1 x k and k x 1 inception style

Help reduce memory footprint

training: bool