config.json

{
  "model": {
    "variant": "vit_small_patch16_384.augreg_in21k_ft_in1k",
    "input_size": 384,
    "patch_size": 16,
    "freeze_backbone": false,
    "hidden_dropout_prob": 0.0,
    "hidden_size": 384,
    "num_attention_heads": 6,
    "num_hidden_layers": 12,
    "attention_probs_dropout_prob": 0.0,
    "layer_norm_eps": 1e-6,
    "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
  },
  "device": "cuda",
  "dtype": "float32",
  "checkpoint_path": "pretrained_weights/model_v11_ViT_384_base_ckpt.pt"
}