Why my model returns nan?

This might sound weird but restart your machine. I was facing some issue with the GPU and had to restart the system and to my surprise, it started training.

1 Like