models/model_config.json

{
  "model_name": "nnunet",
  "n_channels": 4,
  "n_classes": 5,
  "deep_supervision": true,
  "pocket": true,
  "patch_size": [
    128,
    128,
    128
  ],
  "target_spacing": [
    1.0,
    1.0,
    1.0
  ],
  "use_res_block": true
}