pytorch - RuntimeError : Expected object of type torch. DoubleTensor 但发现参数 #2 的类型为 torch.FloatTensor 'weight'

Question

我的输入张量是 torch.DoubleTensor 类型。但我得到了下面的 RuntimeError:

RuntimeError: Expected object of type torch.DoubleTensor but found type torch.FloatTensor for argument #2 'weight'

我没有明确指定权重的类型(即我没有自己初始化权重。权重是由 pytorch 创建的)。什么会影响正向过程中的权重类型？

非常感谢!!

Answer 1

我也收到完全相同的错误。根本原因原来是我的数据加载代码中的这条语句:

t = t.astype(np.float)

这里 np.float 转换为映射到 DoubleTensor 的 64 位浮点数。所以把这个改成，

t = t.astype(np.float32)

解决了这个问题。