This repository contains the code implementation of the FeedforwardNeuralNetwork used to train and test the MNIST dataset in PyTorch.
Steps to run:
In your terminal, run the following commands:
- pip install torch torchvision numpy matplotlib
- git clone https://github.com/sugheert/DL_Assignment1 sugheert_DL_a1
Open the file sugheert_DL_a1/DL_Assignment1_Q4_2.ipynb in any IDE with python and jupyter notebook support and give 'Run All' option.
Through Google Colab:
- Upload the 'DL_Assignment1_Q4_2.ipynb' file to Google Colab
- Add the following cell at the top: ' pip install torch torchvision numpy matplotlib '
- Run all cells