NaN or Inf found in input tensor

[6zhenpi]

Hello Author, I'm glad to run your code, but I encountered an issue as shown in the image below. Could you please advise how to resolve it?

[YunHDan]

i used origin settings to run the code "python code/train_stage2.py" and got a same problem about "Nan or Inf found in input tensor" causing by nll is a nan value. How to solve that problem?