RuntimeError: expected a Variable argument, but got torch.FloatTensor

从字面的意思来看是

RuntimeError:期望有一个变量参数,但是得到了torch.FloatTensor

这个解决方法是升级pytorch的版本就可以了,我之前用的是pytorch0.3,现在升级了就不会出现这个问题了。

出现问题的代码

import torch
from torch import nn

m = nn.AdaptiveMaxPool2d(1)
input = torch.randn(1, 64, 8, 9)
output = m(input)
print(output.size())

想要做一下关于nn.AdaptiveMaxPool2d()的测试,这个方法一般使用的是在神经网络里面,所以输入参数都是变量。然后我使用固定值,所以会出现问题。然后pytorch将这个函数做了升级,现在可以了

相关推荐
©️2020 CSDN 皮肤主题: 终极编程指南 设计师:CSDN官方博客 返回首页