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.utils.data
- PyTorch Documentation, PyTorch Foundation, 2025 - Official documentation for PyTorch's data loading utilities, including Dataset
and DataLoader
, which support torchvision
datasets.torchvision.datasets
- PyTorch Documentation, PyTorch Foundation, 2024 - Official documentation describing the built-in datasets available in TorchVision, their usage, and configuration parameters.Dataset
, DataLoader
, and working with torchvision
datasets effectively.