File size: 978 Bytes
ef5c55a | 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 | {
"model": {
"type": "PT-v3m2",
"in_channels": 4,
"order": [
"hilbert",
"hilbert-trans",
"z",
"z-trans"
],
"stride": [
2,
2,
2,
2
],
"enc_depths": [
3,
3,
3,
9,
3
],
"enc_channels": [
48,
96,
192,
384,
512
],
"enc_num_head": [
3,
6,
12,
24,
32
],
"enc_patch_size": [
256,
256,
256,
256,
256
],
"mlp_ratio": 4,
"qkv_bias": true,
"qk_scale": null,
"layer_scale": 1e-05,
"attn_drop": 0.0,
"proj_drop": 0.0,
"drop_path": 0.3,
"shuffle_orders": true,
"pre_norm": true,
"enable_rpe": false,
"enable_flash": true,
"upcast_attention": false,
"upcast_softmax": false,
"traceable": true,
"enc_mode": true,
"mask_token": true,
"cpe_first_layer_only": false,
"cpe_shared_weight": false
}
}
|