{ "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" }