code: def get_configs(model_name: str = None):
      configs = {
          "img_size": 224,
          "num_classes": 1000,
          "embed_dims": [64, 128, 256, 512],
          "num_heads": [2, 4, 8, 16],
          "mlp_ratios": [8, 8, 4, 4],
          "qkv_bias": True,
          "qk_scale": None,
          "drop_rate": 0.0,
          "attn_drop_rate": 0.0,
          "drop_path_rate": 0.0,
          "norm_layer": partial(nn.LayerNorm, eps=1e-6),
          "depths": [3, 4, 24, 2],
          "sr_ratios": [8, 4, 2, 1],
          "num_stages": 4,
          "num_conv": 2,
      }
  
      if model_name is None:
          pass
      elif model_name == "shunted_t":
          configs.update(depths=[1, 2, 4, 1], num_conv=0)
      elif model_name == "shunted_s":
          configs.update(depths=[2, 4, 12, 1], num_conv=1)
      elif model_name == "shunted_b":
          pass
      else:
          raise ValueError(f"Invalid model name: {model_name}.")
      return configs
  
imports: from functools import partial
  from torch import nn
  from collections import OrderedDict
label_desc: Use image node_id 1 for calling a global function with example usage: get_configs(model_name) and returns: configs