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
} |