This project implements a simple fully connected neural network (MLP) to classify handwritten digits from the MNIST dataset using PyTorch.
The network consists of four fully connected layers with ReLU activations and is trained using the Adam optimizer.
- Fully connected neural network (MLP) for MNIST
- Uses PyTorch and torchvision
- Visualizes sample predictions
- Outputs test accuracy
- Easy to modify for experimenting with architectures
- Python 3.6+
- PyTorch
- Torchvision
- Matplotlib
Install dependencies using pip:
pip install torch torchvision matplotlib