config.json

{"model_type": "SimpleCNN", "input_size": [3, 224, 224], "output_size": 2}