jrsimuix commited on
Commit
70febf7
1 Parent(s): ca6a497

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +14 -0
  2. preprocessor_config.json +7 -0
config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "vit",
3
+ "hidden_size": 768,
4
+ "num_attention_heads": 12,
5
+ "num_hidden_layers": 12,
6
+ "intermediate_size": 3072,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.1,
9
+ "attention_probs_dropout_prob": 0.1,
10
+ "image_size": 128,
11
+ "patch_size": 16,
12
+ "num_channels": 3,
13
+ "num_labels": 2
14
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "feature_extractor_type": "ViTFeatureExtractor",
3
+ "image_mean": [0.5, 0.5, 0.5],
4
+ "image_std": [0.5, 0.5, 0.5],
5
+ "size": 128
6
+ }
7
+