趋近智
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()
, PyTorch Documentation, 2017 (PyTorch Foundation) - 官方 PyTorch 文档,解释了在评估期间使用 torch.no_grad()
进行内存和计算效率优化的目的和用法,以及 model.eval()
将模型层设置为推理模式。