AbstractPhil commited on
Commit
8cb896f
·
verified ·
1 Parent(s): cf9a64f

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +23 -0
config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "geolip_vit",
3
+ "auto_map": {
4
+ "AutoConfig": "modeling_geolip_vit.GeoLIPViTConfig",
5
+ "AutoModel": "modeling_geolip_vit.GeoLIPViTModel"
6
+ },
7
+ "image_size": 224,
8
+ "patch_size": 16,
9
+ "hidden_size": 384,
10
+ "num_attention_heads": 6,
11
+ "num_hidden_layers": 6,
12
+ "intermediate_size": 1536,
13
+ "output_dim": 128,
14
+ "n_anchors": 256,
15
+ "n_comp": 8,
16
+ "d_comp": 64,
17
+ "n_classes": 80,
18
+ "hidden_dropout_prob": 0.1,
19
+ "soup_enabled": true,
20
+ "consensus_cv": 0.2731,
21
+ "experts": ["clip_l14_openai", "dinov2_b14", "siglip_b16_384"],
22
+ "torch_dtype": "float32"
23
+ }