Skip to content

The Setup

Building towards: MNIST dataset

A case study of the pytorch MNIST example.

  1. Boilerplate: Setting up python and virtual environments
  2. MNIST: Code and breakdown of MNIST example
  3. Randomness: Examples of setting random seed and effect on data, models, etc.
  4. Dataset: Investigation of datasets.MNIST
  5. Model: Investigation of torch.nn.Module
  6. Metrics: MNIST classification metrics
  7. Training: MNIST train function
  8. Device: Ensuring use of GPU for training
  9. Hyperparams: Providing hyperparams to python training script