config.json

{"num_classes": 10, "input_size": [1, 28, 28]}