| |
| norm_cfg = dict(type='SyncBN', requires_grad=True) |
| model = dict( |
| type='EncoderDecoder', |
| backbone=dict( |
| type='VIT_MLA', |
| model_name='vit_large_patch16_384', |
| img_size=768, |
| patch_size=16, |
| in_chans=3, |
| embed_dim=1024, |
| depth=24, |
| num_heads=16, |
| num_classes=19, |
| drop_rate=0.1, |
| norm_cfg=norm_cfg, |
| pos_embed_interp=True, |
| align_corners=False, |
| mla_channels=256, |
| mla_index=(5,11,17,23) |
| ), |
| decode_head=dict( |
| type='VIT_MLAHead', |
| in_channels=1024, |
| channels=512, |
| img_size=768, |
| mla_channels=256, |
| mlahead_channels=128, |
| num_classes=19, |
| norm_cfg=norm_cfg, |
| align_corners=False, |
| loss_decode=dict( |
| type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0))) |
| |
| train_cfg = dict() |
| test_cfg = dict(mode='whole') |
|
|
|
|
|
|
