Code for the Deep Learning with PyTorch lesson - View it on GitHub