geolip-vit-base-x3 / config.json
AbstractPhil's picture
Create config.json
8cb896f verified
raw
history blame contribute delete
575 Bytes
{
"model_type": "geolip_vit",
"auto_map": {
"AutoConfig": "modeling_geolip_vit.GeoLIPViTConfig",
"AutoModel": "modeling_geolip_vit.GeoLIPViTModel"
},
"image_size": 224,
"patch_size": 16,
"hidden_size": 384,
"num_attention_heads": 6,
"num_hidden_layers": 6,
"intermediate_size": 1536,
"output_dim": 128,
"n_anchors": 256,
"n_comp": 8,
"d_comp": 64,
"n_classes": 80,
"hidden_dropout_prob": 0.1,
"soup_enabled": true,
"consensus_cv": 0.2731,
"experts": ["clip_l14_openai", "dinov2_b14", "siglip_b16_384"],
"torch_dtype": "float32"
}