趋近智
requires_grad
)backward()
).grad
)torch.nn
搭建模型torch.nn.Module
基类torch.nn
损失)torch.optim
)torch.utils.data.Dataset
torchvision.transforms
)torch.utils.data.DataLoader
这部分内容有帮助吗?
torch.no_grad()
上下文管理器的官方文档,详细说明了其在推理和评估期间梯度跟踪的用法和影响。.detach()
方法的官方文档,解释了如何创建一个与计算图分离的新张量。