A simple Convolutional Neural Network (CNN) built with PyTorch to classify handwritten digits from the MNIST dataset. Perfect for learning the basics of deep learning, computer vision, and PyTorch.
- Custom CNN architecture with 2 convolutional layers and 2 fully connected layers
- Trains on the MNIST dataset (60,000 training images)
- Evaluates test accuracy on 10,000 images
- Uses Adam optimizer and CrossEntropyLoss
pip install -r requirements.txt
python train.py