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
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