Getting nan after few iteration

[PRAYALANKAR]

Hi
I am trying to run the code but after few iterations i am getting the nan in the loss. Apart from the data paths i have kept everything same.
Any suggestions ?

[mafamily2496]

It's easy to get nan, start another train after getting nan

[Haoxl1997]

I have the same problem

[Wujielun]

I have the same problem

Hi HaoSir, I also meet the problem, do you have any solution now?