3 ways of creating a neural network in PyTorch

Goal¶

This post aims to introduce 3 ways of how to create a neural network using PyTorch:

Three ways:

• nn.Module
• nn.Sequential
• nn.ModuleList

Reference