【发布时间】:2021-09-13 08:51:11
【问题描述】:
PyTorch 版本:1.4.0+cu100 运行时
val_L= Variable(torch.from_numpy(val_L.copy()), volatile=True).cuda()
发生:
UserWarning: volatile was removed and now has no essfet. Use 'with torch.no_grad():' insted.
那么,我应该如何修改代码呢? 非常感谢!!!
【问题讨论】:
-
我将代码修改为如下形式: with torch.no_grad(): val_L = torch.from_numpy(val_L.copy()).cuda 是不是?
标签: python pytorch volatile user-warning