Robots Atlas>ROBOTS ATLAS

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.