Problem description:
This error occurs when LSTM is used for data training, I convert the numpy data directly to the tensor data type in the torch:
RuntimeError: expected scalar type Double but found Float
Cause analysis:
The data type of the tensor is incorrect
x_train_tensor = torch.from_numpy(x_train)
y_train_tensor = torch.from_numpy(y_train)
Solution:
Convert the original tensor to the torch.float32 type
x_train_tensor = torch.from_numpy(x_train).to(torch.float32)
y_train_tensor = torch.from_numpy(y_train).to(torch.float32)