mvit_v2_s_yellowstone.json

{
  "birder_version": "v0.1.1",
  "name": "mvit_v2",
  "alias": "mvit_v2_s",
  "task": "image_classification",
  "model_config": {
    "depths": [
      1,
      2,
      11,
      2
    ],
    "embed_dim": 96,
    "base_heads": 1,
    "dim_mul_in_att": true,
    "use_cls_token": false,
    "drop_path_rate": 0.1
  },
  "signature": {
    "inputs": [
      {
        "data_shape": [
          0,
          3,
          384,
          384
        ]
      }
    ],
    "outputs": [
      {
        "data_shape": [
          0,
          330
        ]
      }
    ]
  },
  "rgb_stats": {
    "mean": [
      0.5191,
      0.5306,
      0.4877
    ],
    "std": [
      0.2316,
      0.2304,
      0.2588
    ]
  }
}