I am training a simple polynomial Model w2 * t_u ** 2 + w1 * t_u + b.
Implementation Details
image874×717 54.5 KB
After few epochs, the loss tends to inf and parameters move to nan as in below image
image872×755 81.6 KB
Can anyone explain why it happens and how to avoid it ?