【发布时间】:2020-07-07 21:48:08
【问题描述】:
我遇到了这个错误,你能帮帮我吗? 这是我在 utils.py 文件中的错误
content/gdrive/My Drive/DeepFakeDetection/utils.py:21: UserWarning:
volatile was removed and now has no effect. Use with torch.no_grad():
instead. return Variable(x, volatile=volatile)
这是我在 utils.py 文件中的代码
def to_var(x, volatile=False):
if torch.cuda.is_available():
x = x.cuda()
return Variable(x, volatile=volatile)
谢谢
【问题讨论】:
-
Variable和volatile都已弃用,取而代之的是torch.Tensor和上下文管理器torch.no_grad()。无论如何,这只是一个警告而不是错误,因此如果代码运行正确,您可以选择忽略它。