shauray commited on
Commit
1b19a7d
1 Parent(s): 6b796b4

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +22 -0
config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "img_size" : (256, 192),
7
+ "patch_size" : 16,
8
+ "embed_dim" : 384,
9
+ "depth" : 12,
10
+ "num_heads" : 12,
11
+ "ratio" : 1,
12
+ "use_checkpoint" : False,
13
+ "mlp_ratio" : 4,
14
+ "qkv_bias" : True,
15
+ "drop_path_rate" : .1,
16
+ "keypoint_in_channels" : 382,
17
+ "keypoint_num_deconv_layers" : 2,
18
+ "keypoint_num_deconv_filters" : (256, 256),
19
+ "keypoint_num_deconv_kernels" : (4,4),
20
+ "dropout_p" : 0.0,
21
+ "num_output_channels" : 17,
22
+ }