config.json

{
    "model_name": "mnist_classifier",
    "input_size": [1, 28, 28],
    "hidden_size": 128,
    "num_classes": 10,
    "dropout": 0.5,
    "mean": 0.1307,
    "std": 0.3081
}