File size: 1,654 Bytes
61009ee | 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 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | {
"model": {
"type": "DefaultSegmentorV2",
"num_classes": 5,
"backbone_out_channels": 64,
"freeze_backbone": false,
"backbone": {
"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
],
"dec_depths": [
2,
2,
2,
2
],
"dec_channels": [
64,
96,
192,
384
],
"dec_num_head": [
4,
6,
12,
24
],
"dec_patch_size": [
256,
256,
256,
256
],
"mlp_ratio": 4,
"qkv_bias": true,
"qk_scale": null,
"layer_scale": 0.0,
"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": false,
"mask_token": false,
"enc_mode": false,
"freeze_encoder": false
},
"criteria": [
{
"type": "CrossEntropyLoss",
"loss_weight": 1.0,
"ignore_index": -1
}
]
}
}
|