config.json

{
    "architecture": "tf_efficientnet_b0.ap_in1k",
    "input_size": [
        3,
        384,
        384
    ],
    "num_classes": 1604
}