config.json

{
    "architecture": "attention_unet_model.h5",
    "input_shape": [
        128,
        128,
        2
    ],
    "num_classes": 4,
    "optimizer": "adam",
    "loss": "categorical_crossentropy",
    "metrics": [
        "accuracy"
    ]
}