-
Notifications
You must be signed in to change notification settings - Fork 24.7k
nn Basics
Manuel edited this page Jul 3, 2024
·
5 revisions
- Understand what
torch.nnis - Understand what a module is
- Understand how modules are used to build and train neural networks
- Understand how to author a module in PyTorch
- Understand how to test modules in PyTorch
torch.nn is the component of PyTorch that provides building blocks for neural networks. Its core abstraction is nn.Module, which encapsulates stateful computation with learnable parameters. Modules integrate with the autograd system and are generally trained using optimizers provided in torch.optim.
Read through the following links:
Work through the lab.
Unit 5: torch.nn - C++ API quick code walkthrough
I would love to contribute to PyTorch!