It is ok.

But what is your avg loss?

If it’s nan please stop otherwise continue..