code: def __init__(self, drop_path_rate=0.1, **kwargs):
      super(mit_b3_avg, self).__init__(
          patch_size=4,
          embed_dims=[64, 128, 320, 512],
          num_heads=[1, 2, 5, 8],
          mlp_ratios=[4, 4, 4, 4],
          qkv_bias=True,
          norm_layer=partial(LayerNorm, eps=1e-6),
          depths=[3, 4, 18, 3],
          sr_ratios=[8, 4, 2, 1],
          drop_rate=0.0,
          drop_path_rate=drop_path_rate,
          output_avg=True,
      )
  
imports: import math
  import torch
  import torch.nn
  import torch.nn.functional
  from functools import partial
  from torch.nn.init import trunc_normal_
  from megatron.model.transformer import DropPath
  from megatron.model import LayerNorm
label_desc: Use image node_id 1 to create a new mit_b3_avg object from inherited base classes: MixVisionTransformer with example: obj = mit_b3_avg(drop_path_rate)