isanet_r18-d8_512x512_80k_ade20k.py

_base_ = [
    '../../_base_/models/isanet_r50-d8.py',
    '../../_base_/datasets/mmseg/ade20k.py',
    '../../_base_/mmseg_runtime.py',
    '../../_base_/schedules/mmseg/schedule_80k.py',
]

model = dict(
    pretrained='open-mmlab://resnet18_v1c',
    backbone=dict(depth=18),
    decode_head=dict(
        num_classes=150,
        in_channels=512,
        channels=128
    ),
    auxiliary_head=dict(
        num_classes=150,
        in_channels=256,
        channels=64
    )
)