Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

Training My Neural Network with the MNIST Dataset

Aakash Pydi

The neural network with forward propagation and back-propagation pass is implemented in The MyImg2Num neural network is defined in It relies on the Neural Network class implemented in for its functionality. The NnImg2Num neural network is defined in This neural network uses the torch.nn neural network package for its functionality. Finally both of these neural networks are trained and tested on the MNIST handwritten digits dataset. In order to run the scripts, simply create an instance of the corresponding class and call the train() instance method as shown below.

nn_mnist = NnImg2Num()


my_mnist = MyImg2Num()


The performance charts associated with each of the neural networks are given below.

NnImg2Num Neural Network



MyImg2Num Neural Network



You can’t perform that action at this time.