File size: 1,026 Bytes
ff25d36
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
{
  "attn_drop": 0.0,
  "dec_channels": [
    96,
    96,
    192,
    384
  ],
  "dec_depths": [
    3,
    3,
    3,
    3
  ],
  "dec_num_head": [
    6,
    6,
    12,
    32
  ],
  "dec_patch_size": [
    1024,
    1024,
    1024,
    1024,
    1024
  ],
  "drop_path": 0.3,
  "enable_flash": true,
  "enable_rpe": false,
  "enc_channels": [
    48,
    96,
    192,
    384,
    512
  ],
  "enc_depths": [
    3,
    3,
    3,
    12,
    3
  ],
  "enc_mode": true,
  "enc_num_head": [
    3,
    6,
    12,
    24,
    32
  ],
  "enc_patch_size": [
    1024,
    1024,
    1024,
    1024,
    1024
  ],
  "freeze_encoder": false,
  "in_channels": 9,
  "layer_scale": null,
  "mask_token": true,
  "mlp_ratio": 4,
  "order": [
    "z",
    "z-trans",
    "hilbert",
    "hilbert-trans"
  ],
  "pre_norm": true,
  "proj_drop": 0.0,
  "qk_scale": null,
  "qkv_bias": true,
  "shuffle_orders": true,
  "stride": [
    2,
    2,
    2,
    2
  ],
  "traceable": true,
  "upcast_attention": false,
  "upcast_softmax": false
}