MNIST_Pytorch_Lightning

A PyTorch Lightning example on MNIST Dataset

Requirements

These libraries are required -

PyTorch
scikit-learn
PyTorch Lightning

Change the hyperparameters in run.sh and execute it using the following command:

bash run.sh

The confusion matrix will be stored in the folder visualizations and the logs will be stored in tensorboard_logs.

To start tensorboard run the following command:

tensorboard --logdir='tensorboard_logs'

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
dataloaders		dataloaders
models		models
utils		utils
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
run.sh		run.sh
train.py		train.py