PyTorch Tutorial for Beginners - Building Neural Networks

Essentially, when you want to build some model using PyTorch you can inherit nn.Module class. This way you can just by overriding several methods create different types of neural networks. This is one of the main reasons why PyTorch is so popular in the research community because it gives you "precooked" solutions with enough flexibility. We utilize that to create a model that receives several parameters through the constructor. It receives input size (ie.