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

config = dict(
    model=dict(
        type='AnyUNet',
        params=dict(
            encoder_name='resnet50',
            classes=7,
            encoder_weights='imagenet',
            loss=dict(
                ce=dict()
            )
        )
    ),
    data=data,
    optimizer=optimizer,
    learning_rate=learning_rate,
    train=train,
    test=test
)
