File size: 451 Bytes
245b43a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
_base_ = ['../../../configs/segformer/segformer_mit-b5_8xb1-160k_cityscapes-1024x1024.py']
model = dict(
backbone=dict(
tome_cfg=[
dict(kv_mode='bsm', kv_r=0.6, kv_sx=2, kv_sy=2),
dict(kv_mode='bsm', kv_r=0.6, kv_sx=2, kv_sy=2),
dict(q_mode='bsm', q_r=0.8, q_sx=4, q_sy=4),
dict(q_mode='bsm', q_r=0.8, q_sx=4, q_sy=4)
]
)
)
train_dataloader = dict(batch_size=4, num_workers=8)
|