Convert Your Data to float instead … as it’s very dangerous to convert model type to Double … 62 as GPUs are not good at double precision computation
Try:
X_train = torch.FloatTensor(X_train)

instead