YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

MNIST MLP Classifier (PyTorch)

This is a simple multi-layer perceptron (MLP) trained on the MNIST dataset to classify handwritten digits (0โ€“9).

Model

  • Input: 28ร—28 grayscale image
  • Architecture: MLP (784 โ†’ 128 โ†’ 64 โ†’ 10)
  • Loss: CrossEntropyLoss
  • Optimizer: Adam

Usage

Load weights using:

model.load_state_dict(torch.load("mnist_mlp.pth"))
model.eval()
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support