How can i convert model's parameter data type to Double?

Convert Your Data to float instead … as it’s very dangerous to convert model type to Double … as GPUs are not good at double precision computation
Try:

X_train = torch.FloatTensor(X_train)

instead

1 Like