config.json

{
  "num_classes": 1,
  "input_channels": 1,
  "image_size": [
    512,
    512
  ],
  "features": [
    64,
    64,
    128
  ],
  "attention_dims": [
    64,
    32,
    16
  ],
  "patch_size": 256,
  "batch_size": 4,
  "architectures": [
    "BinarySegmentation"
  ],
  "auto_map": {
    "AutoConfig": "ModelConfiguration.VesselSegmentConfig",
    "AutoModel": "Model.VesselSegmentModel"
  }
}