How to fix RuntimeError “Expected object of scalar type Float but got scalar type Double for argument”?

Question

I\'m trying to train a classifier via PyTorch. However, I am experiencing problems with training when I feed the model with training data. I get this error on y_pred =

Answer 1

The issue can be fixed by setting the datatype of input to Double i.e torch.float32

I hope the issue came because your datatype is torch.float16