PyTorch does contain some ‘off the shelf’ networks ready for use, but here we will look at how to build a simple convolutional nearal network from scratch.
In the linked Colab notebook we will look at the following:
objects and classes
the PyTorch module class
defining trainable layers, such as convolutional layers
adding other layers, such as max pooling layers, and adding activation functions
linking layers and activation functions together into a network
how to intialise run a forward pass through the network.
Follow the link below and work through the Colab notebook step by step.