class_docstr: This class implements the `MobileViT architecture <https://arxiv.org/abs/2110.02178?context=cs.LG>`_
code: def add_arguments(
      cls, parser: argparse.ArgumentParser
  ) -> argparse.ArgumentParser:
      group = parser.add_argument_group(title=cls.__name__)
      group.add_argument(
          "--model.classification.mit.mode",
          type=str,
          default="small",
          choices=["xx_small", "x_small", "small"],
          help="MobileViT mode. Defaults to small",
      )
      group.add_argument(
          "--model.classification.mit.attn-dropout",
          type=float,
          default=0.0,
          help="Dropout in attention layer. Defaults to 0.0",
      )
      group.add_argument(
          "--model.classification.mit.ffn-dropout",
          type=float,
          default=0.0,
          help="Dropout between FFN layers. Defaults to 0.0",
      )
      group.add_argument(
          "--model.classification.mit.dropout",
          type=float,
          default=0.0,
          help="Dropout in Transformer layer. Defaults to 0.0",
      )
      group.add_argument(
          "--model.classification.mit.transformer-norm-layer",
          type=str,
          default="layer_norm",
          help="Normalization layer in transformer. Defaults to LayerNorm",
      )
      group.add_argument(
          "--model.classification.mit.no-fuse-local-global-features",
          action="store_true",
          help="Do not combine local and global features in MobileViT block",
      )
      group.add_argument(
          "--model.classification.mit.conv-kernel-size",
          type=int,
          default=3,
          help="Kernel size of Conv layers in MobileViT block",
      )
  
      group.add_argument(
          "--model.classification.mit.head-dim",
          type=int,
          default=None,
          help="Head dimension in transformer",
      )
      group.add_argument(
          "--model.classification.mit.number-heads",
          type=int,
          default=None,
          help="Number of heads in transformer",
      )
      return parser
  
file_path: ml-cvnets/cvnets/models/classification/mobilevit.py
imports: import argparse
  from typing import Dict, Optional, Tuple
  from torch import nn
  from cvnets.layers import ConvLayer2d, Dropout, GlobalPool, LinearLayer
  from cvnets.models import MODEL_REGISTRY
  from cvnets.models.classification.base_image_encoder import BaseImageEncoder
  from cvnets.models.classification.config.mobilevit import get_configuration
  from cvnets.modules import InvertedResidual, MobileViTBlock
  from utils import logger
label_desc: Use image node_id 2 for calling the MobileViT obj's underlying member method code with example usage: obj.add_arguments(cls, parser) and returns: parser