ML Reproducibility Challenge note:

Because of the filesize limit of 100MB, the pretrained models as well as the MNIST and FMNIST datasets were omitted from the submission.
In order to run the code, they will have to be downloaded manually.
The folder structure should look like this:

datasets
  FashionMNIST/
    processed/
      test.pt
      training.pt
    raw/
      t10k-images-idx3-ubyte
      t10k-labels-idx1-ubyte
      train-images-idx3-ubyte
      train-labels-idx1-ubyte
  MNIST/
    processed/
      test.pt
      training.pt
    raw/
      t10k-images-idx3-ubyte
      t10k-labels-idx1-ubyte
      train-images-idx3-ubyte
      train-labels-idx1-ubyte