趋近智
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.Dataset
这部分内容有帮助吗?
Dataset
及相关类的API参考。Dataset
并使用DataLoader
。torch.utils.data.Dataset
和DataLoader
处理数据的详细说明和示例。