config.json

{
    "model_type": "unet",
    "attention_type": "2d",
    "input_size": 128,
    "volume_slices": 100,
    "volume_start_at": 22,
    "num_classes": 4,
    "activation": "softmax",
    "pipeline_tag": "image-segmentation"
}