Help loss go to infinite

[Enigmo109]

Hello,
I'm trying to fit the model with dataset that u linked on readme. During my test loss value starts at 200 - 800 (depend on batch size and lr) and go to infinite in 2-4 steps.

Hope to receive soon a feedback,
best regards

[Owaiskhan9654]

Check out this closed issue @Enigmo109

#3 (comment)