| { |
| "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" |
| } |
