requires_grad
)backward()
).grad
)torch.nn
torch.nn.Module
Base Classtorch.nn
losses)torch.optim
)torch.utils.data.Dataset
torchvision.transforms
)torch.utils.data.DataLoader
Was this section helpful?
torch.nn
module, which provides the building blocks for neural networks, including nn.Conv2d
, nn.MaxPool2d
, nn.ReLU
, and nn.Linear
layers used in the example.torch.nn.Module
and the layers discussed in the section, reinforcing practical implementation.