Hi,
I am trying to train the model on mixed precision, so for the same I am using the command:
model.half()
But I am getting the following error:
image1363×422 27.8 KB
So when I convet my input and labels also to half but it seem like the error is caused due to the line:
loss.backward()

I have tried to convert loss back into floating point value and run the same, but still I am getting the following error:
image1383×405 41.4 KB
Any suggetions?