config.json

{
  "architectures": [
    "MNIST_Classifier"
  ],
  "auto_map": {
    "AutoConfig": "config.MNIST_config",
    "AutoModel": "model.MNIST_Classifier"
  },
  "hidden_size1": 128,
  "hidden_size2": 64,
  "input_size": 784,
  "model_type": "MNIST_Classifier",
  "output_size": 10,
  "torch_dtype": "float32",
  "transformers_version": "4.35.2"
}