{  
  "architectures": [
    "ViTForImageClassification"
  ],
  "model": {
    
    "variant": "vit_small_patch16_384.augreg_in21k_ft_in1k",
    "input_size": 384,
    "patch_size": 16,
    "freeze_backbone": false,
    "hidden_dropout_prob": 0.0,
    "attention_probs_dropout_prob": 0.0,
    "num_classes": 1,
    "head": {
      "in_features": 384,
      "out_features": 1,
      "bias": true
    }
  },
  "preprocessing": {
    "norm_mean": [0.48145466, 0.4578275, 0.40821073],
    "norm_std": [0.26862954, 0.26130258, 0.27577711],
    "resize_size": 440,
    "crop_size": 384
  },
  "model_type": "vit",
  "dtype": "float32",
  "checkpoint_path": "pretrained_weights/model_v11_ViT_384_base_ckpt.pt"
}