I am getting this error and have no clue how to fix this.
I have MSELoss output and when I call backward() function, I get the error saying ‘Found dtype Double but expected Float’. I’ve seen some post saying I can change my parameters of my model to accept certain dtypes (which I did to float) and I have checked my input dtype (which is torch.cuda.FloatTensor).
What can I do to fix this error??