Transformer from Scratch · PyTorch for Sequence Models
`nn.Module` and Building Model Blocks
PyTorch for Sequence Models
Introduction
nn.Module is the basic building unit of PyTorch models. In this lesson you will learn about parameters, submodules, the forward method, ModuleList, buffers and train/eval modes.