eugenehp commited on
Commit
24c1f61
·
verified ·
1 Parent(s): 834d93c

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +44 -0
config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "models": {
3
+ "small": {
4
+ "patch_size": 16,
5
+ "embed_dim": 512,
6
+ "depth": 8,
7
+ "num_heads": 8,
8
+ "mlp_ratio": 4.0,
9
+ "decoder_embed_dim": 384,
10
+ "decoder_depth": 4,
11
+ "decoder_num_heads": 16
12
+ },
13
+ "base": {
14
+ "patch_size": 16,
15
+ "embed_dim": 768,
16
+ "depth": 12,
17
+ "num_heads": 12,
18
+ "mlp_ratio": 4.0,
19
+ "decoder_embed_dim": 512,
20
+ "decoder_depth": 8,
21
+ "decoder_num_heads": 16
22
+ },
23
+ "large": {
24
+ "patch_size": 16,
25
+ "embed_dim": 1024,
26
+ "depth": 24,
27
+ "num_heads": 16,
28
+ "mlp_ratio": 4.0,
29
+ "decoder_embed_dim": 512,
30
+ "decoder_depth": 8,
31
+ "decoder_num_heads": 16
32
+ },
33
+ "largeV2": {
34
+ "patch_size": 16,
35
+ "embed_dim": 1024,
36
+ "depth": 24,
37
+ "num_heads": 16,
38
+ "mlp_ratio": 4.0,
39
+ "decoder_embed_dim": 512,
40
+ "decoder_depth": 8,
41
+ "decoder_num_heads": 16
42
+ }
43
+ }
44
+ }