pytorch
测试时内存暴增，考虑前向传播时梯度被保存，torch.no_grad()要在变量创建之前使用，变量已创建使用无效。
#关闭梯度
for param in model.parameters():
param.requires_grad=False
那些年，踩过的坑
博客指出PyTorch测试时会出现内存暴增问题，原因是前向传播时梯度被保存。解决办法是使用torch.no_grad()，且要在变量创建之前使用，若变量已创建则使用无效。
博客指出PyTorch测试时会出现内存暴增问题，原因是前向传播时梯度被保存。解决办法是使用torch.no_grad()，且要在变量创建之前使用，若变量已创建则使用无效。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
