The Reproducible MNIST

Reproducible MNIST experiments in PyTorch.

Setup

$ python -m venv .venv
$ source .venv/bin/activate
$ pip install --upgrade pip
$ pip install -r requirements.txt
$ wandb login API_KEY

Model Training

$ CUBLAS_WORKSPACE_CONFIG=:16:8 python train.py  # or CUBLAS_WORKSPACE_CONFIG=:16:8 ./train.py

You should set the environment variable CUBLAS_WORKSPACE_CONFIG according to CUDA documentation.

Bonus: WandB takes care of generating the loss plot and saving the training configuration.

Model Evaluation

$ python eval.py  # or ./eval.py

License

MIT License