Pytorch中变量的volatile属性是什么?这是一个用于在PyTorch中定义变量的示例代码 .
datatensor = Variable(data, volatile=True)
基本上,如果您仅进行推理,则将网络的输入设置为易失性,并且不会运行反向传播以节省内存 .
来自docs:
当你确定你甚至不会调用.backward()时,建议使用Volatile进行纯推理模式 . 它比任何其他autograd设置更有效 - 它将使用绝对最小的内存量来评估模型 . volatile也确定requires_grad为False .
编辑:从pytorch版本0.4.0开始,volatile关键字一直是deprecated
1 回答
基本上,如果您仅进行推理,则将网络的输入设置为易失性,并且不会运行反向传播以节省内存 .
来自docs:
编辑:从pytorch版本0.4.0开始,volatile关键字一直是deprecated