config.json

{
    "model_type": "UNet",
    "encoder_name": "resnet34",
    "encoder_weights": "imagenet",
    "in_channels": 3,
    "classes": 3,
    "input_resolution": "{image_size}{image_size}"
}