Building The Discriminator

The Discriminator class

Let’s code the discriminator. Just like before, it is a neural network with a class inherited from nn.Module. We follow the expected PyTorch patterns for initializing the network and providing a forward() function.

Have a look at the following constructor for a Discriminator class.