class_docstr: MobileOne building block.
  
  This block has a multi-branched architecture at train-time
  and plain-CNN style architecture at inference time
  For more details, please refer to our paper:
  `An Improved One millisecond Mobile Backbone` -
  https://arxiv.org/pdf/2206.04040.pdf
class_docstr_tok: ["MobileOne","building","block",".","This","block","has","a","multi-branched","architecture","at","train-time","and","plain-CNN","style","architecture","at","inference","time","For","more","details",",","please","refer","to","our","paper",":","`","An","Improved","One","millisecond","Mobile","Backbone","`","-","https",":","\/\/arxiv.org\/pdf\/2206.04040.pdf"]
docstr: Helper method to construct conv-batchnorm layers.
  
  Args:
      kernel_size: Size of the convolution kernel.
      padding: Zero-padding size.
  
  Returns:
      Conv-BN module.
docstr_tok: ["Helper","method","to","construct","conv-batchnorm","layers",".","Args",":","kernel_size",":","Size","of","the","convolution","kernel",".","padding",":","Zero-padding","size",".","Returns",":","Conv-BN","module","."]
code: def _conv_bn(self, kernel_size: int, padding: int) -> nn.Sequential:
      """Helper method to construct conv-batchnorm layers.
  
      Args:
          kernel_size: Size of the convolution kernel.
          padding: Zero-padding size.
  
      Returns:
          Conv-BN module.
      """
      mod_list = nn.Sequential()
      mod_list.add_module(
          "conv",
          nn.Conv2d(
              in_channels=self.in_channels,
              out_channels=self.out_channels,
              kernel_size=kernel_size,
              stride=self.stride,
              padding=padding,
              groups=self.groups,
              bias=False,
          ),
      )
      mod_list.add_module(
          "bn", nn.BatchNorm2d(num_features=self.out_channels)
      )
      return mod_list
  
file_path: ml-fastvit/models/modules/mobileone.py
imports: from typing import Union, Tuple
  import copy
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 6 for calling the MobileOneBlock obj's underlying member method code with example usage: obj._conv_bn(kernel_size, padding) and returns: mod_list