towhee.models.utils.create_conv2d.create_conv2d

towhee.models.utils.create_conv2d.create_conv2d(in_channels: int, out_channels: int, kernel_size: int, **kwargs) → Module

Select a 2d convolution implementation based on arguments Creates and returns one of torch.nn.Conv2d, Conv2dSame, MixedConv2d, or CondConv2d. Used extensively by EfficientNet, MobileNetv3 and related networks.