- 博客(3)
- 收藏
- 关注
RSS订阅原创 with torch.no_grad():
在深度学习中，我们通过反向传播来计算模型参数的梯度。但是，在进行测试集验证或者模型推断时，并不需要计算梯度，这时计算梯度只会浪费计算资源。因此，为了提高运行效率，可以使用。上下文管理器时，该代码块内的所有操作都将被视为不需要梯度计算，包括所有的前向计算，反向传播和参数更新等操作。这样可以节省显存和计算资源，并且减少计算时间。是PyTorch中的上下文管理器，用于指定一段代码块内部不需要计算梯度。通常在测试集验证或者模型推断时使用。上下文管理器来指定这段代码块内部不需要计算梯度。
2023-05-02 20:39:48
1007
原创 x.grad.zero_()报错‘NoneType‘ object has no attribute ‘zero_‘
【代码】x.grad.zero_()报错‘NoneType‘ object has no attribute ‘zero_‘
2023-05-02 18:44:09
478
1
原创 envs\pytorch-gpu\lib\site-packages\torchvision\io\image.py:13: UserWarning: Failed to load image
pytorch，torchvision，python 三者的对应关系来源于官方 github，链接如下：https://github.com/pytorch/vision#installation。
2023-05-01 09:26:08
412
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人