class_docstr: This class implements `FastViT architecture <todo: add arxiv link here>`_
class_docstr_tok: ["This","class","implements","`","FastViT","architecture","<","todo",":","add","arxiv","link","here",">","`","_"]
docstr: Add model specific arguments
docstr_tok: ["Add","model","specific","arguments"]
code: def add_arguments(
      cls, parser: argparse.ArgumentParser
  ) -> argparse.ArgumentParser:
      """Add model specific arguments"""
      group = parser.add_argument_group(title=cls.__name__)
      group.add_argument(
          "--model.classification.fastvit.variant",
          type=str,
          default="T8",
          help="Variant string for FastViT. Default: T8",
      )
      group.add_argument(
          "--model.classification.fastvit.inference-mode",
          type=bool,
          default=False,
          help="Flag to instantiate inference mode architecture. Default: False",
      )
      group.add_argument(
          "--model.classification.fastvit.dropout",
          type=float,
          default=0.0,
          help="Dropout rate for regularization. Default: 0.0",
      )
      group.add_argument(
          "--model.classification.fastvit.drop-path",
          type=float,
          default=0.0,
          help="Drop path rate. Default: 0.0",
      )
      group.add_argument(
          "--model.classification.fastvit.use-layer-scale",
          type=bool,
          default=True,
          help="Flag to turn on layer scale regularization. Default: True",
      )
      group.add_argument(
          "--model.classification.fastvit.layer-scale-init-value",
          type=float,
          default=1e-5,
          help="Drop path rate. Default: 1e-5",
      )
      return parser
  
file_path: ml-cvnets/cvnets/models/classification/fastvit.py
imports: import argparse
  import copy
  from typing import List
  import torch.nn
  from cvnets.layers import GlobalPool, Identity, LinearLayer
  from cvnets.models import MODEL_REGISTRY
  from cvnets.models.classification.base_image_encoder import BaseImageEncoder
  from cvnets.models.classification.config.fastvit import get_configuration
  from cvnets.modules.fastvit import AttentionBlock, PatchEmbed, RepMixerBlock, convolutional_stem
  from cvnets.modules.mobileone_block import MobileOneBlock
  from utils import logger
label_desc: Use image node_id 2 for calling the FastViT obj's underlying member method code with example usage: obj.add_arguments(cls, parser) and returns: parser