config.json

{
  "N": 2,
  "catorig": false,
  "n_channels": 1,
  "n_classes": 2,
  "outker": 1,
  "skip": true,
  "skipcat": false,
  "width": 32,
  "model_type": "custom_unet",
  "auto_map": {
    "AutoConfig": "unet.UNetConfig",
    "AutoModel": "unet.UNet"
  }
}