Deep Learning with PyTorch

Building Neural Networks

25 min8 activities0 complete

1 of 8

Explore the idea

nn.Module: The Base Class for All Models

Opening your lesson…

Opening your space…

IN THIS LESSON

YOU’RE LEARNING TO

↗define a custom neural network by subclassing nn.Module and implementing forward()

↗compose linear layers and activation functions to build a feedforward network

↗choose an appropriate loss function for classification or regression tasks

↗instantiate an optimizer and understand what its step() method does

↗inspect and count a model's trainable parameters