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