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

This issue can also occur if the wrong loss function is selected. For example, if you have regression problem, but you are trying to use cross entropy loss. Then it will be fixed by changing your loss function on MSE