RuntimeError: expected mat1 and mat2 to have the same dtype, but got: c10::Half != float

[A-smarneh]

Hi,

Thanks a lot for this notebook. I have followed the notebook and when I get to the

"trainer.train()"

I get the following run time error:

RuntimeError: expected mat1 and mat2 to have the same dtype, but got: c10::Half != float

I am working on AWS SageMaker, with ml.p3.2xlagre instance. I would appreciate any help you can provide.

[Batwho]

same issue, are you able to figure this out?

[A-smarneh]

@Batwho the problem was in the dataset. The dataset contained question and answer field and the model expected 3 question, context and answer

[Batwho]

got it thanks!