class_docstr: This class implements the `MobileViT architecture <https://arxiv.org/abs/2110.02178?context=cs.LG>`_
class_docstr_tok: ["This","class","implements","the","`","MobileViT","architecture","<","https",":","\/\/arxiv.org\/abs\/2110.02178","?","context=cs.LG",">","`","_"]
code: def _make_layer(
      self, input_channel, cfg: Dict
  ) -> Tuple[nn.Sequential, int]:
      block_type = cfg.get("block_type", "mobilevit")
      if block_type.lower() == "mobilevit":
          return self._make_mit_layer(
              input_channel=input_channel, cfg=cfg
          )
      else:
          return self._make_mobilenet_layer(
              input_channel=input_channel, cfg=cfg
          )
  
imports: from typing import Optional, Tuple, Union, Dict
  import math
  import torch
  import torch.nn
  from torch import Tensor
  from torch.nn import functional
  from transformer import TransformerEncoder
  from model_config import get_config
label_desc: Use image node_id 2 for calling the MobileViT obj's underlying member method code with example usage: obj._make_layer(input_channel, cfg) and returns: self, self