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¶
-