HarshRangwala:

xtrainTensor = torch.from_numpy(xtrain_norm)
xtestTensor = torch.from_numpy(xtest_norm)
xvalidTensor = torch.from_numpy(xvalid_norm)



Add .float at the end or use dtype=torch.float as argument for above lines. Also, remove .float for labels, the real reason of first issue.