config.json

{
    "model_type": "UNetTimmWithClassification",
    "input_channels": 3,
    "num_classes": 2,
    "image_size": [224, 224],
    "framework": "pytorch",
    "author": "garyzgao",
    "description": "U-Net model with segmentation functionality for archaeological sherd classification"
}