Expected object of scalar type Double but got scalar type Float for argument #2 ‘mat2’
在写pytorch的时候遇到的坑
把所有模型、变量都变成double
例如
#模型变double
self.rnn = torch.nn.GRU(input_size=32000, hidden_size=24, num_layers=1)
self.rnn = self.rnn.double()
#变量变double
i = i.type(torch.DoubleTensor)

被折叠的 条评论
为什么被折叠?



