from configs.base.loveda import train, test, data, optimizer, learning_rate

config = dict(
    model=dict(
        type='HRNetFusion',
        params=dict(
            backbone=dict(
                hrnet_type='hrnetv2_w40',
                pretrained=True,
                norm_eval=False,
                frozen_stages=-1,
                with_cp=False,
                with_gc=False,
            ),
            neck=dict(
                in_channels=600,
            ),
            classes=7,
            head=dict(
                in_channels=600,
                upsample_scale=4.0,
            ),
            loss=dict(
                ignore_index=-1,
                ce=dict(),
            )
        )
    ),
    data=data,
    optimizer=optimizer,
    learning_rate=learning_rate,
    train=train,
    test=test
)
