DataScience Blog Transfer Learning in PyTorch, Part 2: How to Create a Transfer Learning Class and Train on Kaggle's Test Set

In part 1 of this tutorial, we developed some foundation building blocks as classes in our journey to developing a transfer learning solution in PyTorch. Specifically, we built datasets and DataLoaders for train, validation, and testing using PyTorch API, and ended up building a fully connected class on top of PyTorch's core NN module. We also trained and tested a fully connected model to classify handwritten digits in the famous MNIST dataset and achieved impressive results.