首页 文章

什么是Pytorch中的volatile变量

提问于
浏览
8

Pytorch中变量的volatile属性是什么?这是一个用于在PyTorch中定义变量的示例代码 .

datatensor = Variable(data, volatile=True)

1 回答

  • 12

    基本上,如果您仅进行推理,则将网络的输入设置为易失性,并且不会运行反向传播以节省内存 .

    来自docs

    当你确定你甚至不会调用.backward()时,建议使用Volatile进行纯推理模式 . 它比任何其他autograd设置更有效 - 它将使用绝对最小的内存量来评估模型 . volatile也确定requires_grad为False .

    编辑:从pytorch版本0.4.0开始,volatile关键字一直是deprecated

相关问题