RuntimeError: 标量类型 Double 的预期对象，但参数 #2 得到标量类型 Float

Question

我有一个PyTorch LSTM 模型，我的forward 函数看起来像：

    def forward(self, x, hidden):
        print('in forward', x.dtype, hidden[0].dtype, hidden[1].dtype)
        lstm_out, hidden = self.lstm(x, hidden)
        return lstm_out, hidden

所有print 语句都显示torch.float64，我认为这是一个双倍。那我为什么会遇到这个问题呢？

我已经在所有相关的地方投了double。

Answer 1

确保你的数据和模型都在dtypedouble。

对于模型：

net = net.double()

对于数据：

net(x.double())

一直是discussed on PyTorch forum。