趋近智
requires_grad
)backward()
).grad
)torch.nn
搭建模型torch.nn.Module
基类torch.nn
损失)torch.optim
)torch.utils.data.Dataset
torchvision.transforms
)torch.utils.data.DataLoader
这部分内容有帮助吗?
torch.utils.data
- PyTorch Documentation, PyTorch Foundation, 2025 - PyTorch数据加载工具的官方文档,包括支持torchvision
数据集的Dataset
和DataLoader
。torchvision.datasets
- PyTorch Documentation, PyTorch Foundation, 2024 - 描述TorchVision中内置数据集、其用法和配置参数的官方文档。Dataset
、DataLoader
以及有效使用torchvision
数据集。