Hello,
Yeah I converted my model to double via model.double() and it is working alright now.
But just for curiosity, If I want to convert the data to float, how can I do that.
I am working on regression problem.
I am using sklearn.preprocessing.StandardScaler() for reading the data from csv files which converts by default into float64 aka double.
I also tried to convert the data into float by this:
train_in_data.float()
train_out_data.float()
test_in_data.float()
test_out_data.float()

but it doesn’t convert.
Can you suggest any other solution?
train_in_data, train_out_data, test_in_data, test_out_data are torch tensors.
Even after running the code which I mentioned above it says
torch.tensor(____, dtype=float64)