_base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py' model = dict( pretrained='torchvision://resnet18', backbone=dict(type='ResNet', depth=18), decode_head=dict( in_channels=512, channels=128, ), auxiliary_head=dict(in_channels=256, channels=64))