config.json

{"model_type": "custom", "input_size": [3, 128, 128], "num_classes": 2}