Problem with loss

[HeVLF]

Hello, thank you for sharing your work! I am trying to run your code but I'm encountering some errors with computing the loss, mainly due to the majority of the epochs having a loss equal to "inf" or even "nan". So I am not sure if I changed something without noticing. But would love to hear from you or someone that has faced the same issue.

Best.