Introduction to Deep Learning with PyTorch
Chapter 7: Building and Training a simple Classification Model
Introduction
Let’s put all the previous chapters into practice!
In this chapter, we will provide a step-by-step implementation of a Deep Learning (DL) model, for a basic task: classification of digits in MNIST.
We already used the example of MNIST classification in the previous chapter.
But here, we suggest to start with a simple Deep Learning model, and add new components to it step-by-step.
If you want to implement this model in a Google Colab Notebook, you can use this one