config.json

{
  "_model_class": "PSPNet",
  "activation": null,
  "aux_params": null,
  "classes": 1,
  "decoder_use_norm": "batchnorm",
  "encoder_depth": 3,
  "encoder_name": "timm-efficientnet-b7",
  "encoder_weights": "imagenet",
  "in_channels": 3,
  "psp_dropout": 0.2,
  "psp_out_channels": 512,
  "upsampling": 8
}