File "C:\Users\Armihia\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\nn\modules\linear.py", line 114, in forward return F.linear(input, self.weight, self.bias) RuntimeError: expected scalar type Double but found Float

这个错误通常发生在使用 PyTorch 的线性层（Linear）时，因为 PyTorch 中默认的数据类型是 FloatTensor，而有些模型需要的是 DoubleTensor 类型的数据。解决这个问题的方法是将数据类型转换为 DoubleTensor，可以使用以下代码：
x = x.double() # 将输入 x 转换为 DoubleTensor 类型

# 在定义线性层时，需要指定权重和偏置的数据类型为 DoubleTensor
self.fc = nn.Linear(in_features, out_features, bias=True, dtype=torch.double)
登录后复制
另外，也可以在模型的 forward 方法中将输入数据转换为 DoubleTensor 类型，例如：
class MyModel(nn.Module):
def __init__(self):
super(MyModel, self).__init__()
self.fc1 = nn.Linear(10, 5, bias=True, dtype=torch.double)
self.fc2 = nn.Linear(5, 1, bias=True, dtype=torch.double)

def forward(self, x):
x = x.double() # 将输入 x 转换为 DoubleTensor 类型
x = self.fc1(x)
x = F.relu(x)
x = self.fc2(x)

return x
登录后复制