mnmly commited on
Commit
1c42bac
·
verified ·
1 Parent(s): baa2471

Upload utonia_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. utonia_config.json +59 -0
utonia_config.json ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "in_channels": 9,
3
+ "order": [
4
+ "z",
5
+ "z-trans",
6
+ "hilbert",
7
+ "hilbert-trans"
8
+ ],
9
+ "stride": [
10
+ 2,
11
+ 2,
12
+ 2,
13
+ 2
14
+ ],
15
+ "enc_depths": [
16
+ 3,
17
+ 3,
18
+ 3,
19
+ 12,
20
+ 3
21
+ ],
22
+ "enc_channels": [
23
+ 54,
24
+ 108,
25
+ 216,
26
+ 432,
27
+ 576
28
+ ],
29
+ "enc_num_head": [
30
+ 3,
31
+ 6,
32
+ 12,
33
+ 24,
34
+ 32
35
+ ],
36
+ "enc_patch_size": [
37
+ 1024,
38
+ 1024,
39
+ 1024,
40
+ 1024,
41
+ 1024
42
+ ],
43
+ "mlp_ratio": 4,
44
+ "qkv_bias": true,
45
+ "qk_scale": null,
46
+ "attn_drop": 0.0,
47
+ "proj_drop": 0.0,
48
+ "drop_path": 0.3,
49
+ "shuffle_orders": true,
50
+ "pre_norm": true,
51
+ "enable_rpe": false,
52
+ "enable_flash": true,
53
+ "upcast_attention": false,
54
+ "upcast_softmax": false,
55
+ "traceable": true,
56
+ "enc_mode": true,
57
+ "mask_token": true,
58
+ "rope_base": 10
59
+ }