config.json

{
  "_model_class": "Unet",
  "activation": null,
  "aux_params": null,
  "classes": 1,
  "decoder_attention_type": null,
  "decoder_channels": [
    256,
    128,
    64,
    32,
    16
  ],
  "decoder_use_batchnorm": true,
  "encoder_depth": 5,
  "encoder_name": "resnet34",
  "encoder_weights": "imagenet",
  "in_channels": 3
}