Skip to content

A PyTorch-based CNN that classifies handwritten digits from the MNIST dataset. Features two convolutional and two fully connected layers, trained with Adam optimizer and CrossEntropyLoss. Ideal for learning image classification, deep learning workflows, and neural network fundamentals.

Notifications You must be signed in to change notification settings

Rakza31/mnist_scnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 

Repository files navigation

MNIST Digit Classifier

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.


Features ✨

  • 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

Installation & Usage ⚡

pip install -r requirements.txt
python train.py

About

A PyTorch-based CNN that classifies handwritten digits from the MNIST dataset. Features two convolutional and two fully connected layers, trained with Adam optimizer and CrossEntropyLoss. Ideal for learning image classification, deep learning workflows, and neural network fundamentals.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages