1、可以減少內存使用
2、訓練集訓練好模型后，在驗證集這里使用with torch.no_grad()，訓練集則不會計算梯度值，然后並不會改變模型的參數，只是看了訓練的效果。
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
with torch.no_grad()詳解
Torch.no_grad()影響MSE損失
PyTorch中，關於model.eval()和torch.no_grad()
Pytorch 中 model.eval() 和 with torch.no_grad() 的區別
torch.nn.utils.clip_grad_norm_()
torch.tensor默認requires_grad=False
pytorch中梯度爆炸解決方法 torch.nn.utils.clip_grad_norm_()
[torch] torch.contiguous
model.zero_grad() & optimizer.zero_grad()
數組指針的用法，用處。