config.json

{
  "arch": "unet",
  "classes": 3,
  "encoder": "maxvit_base_tf_512",
  "in_channels": 3
}