[Solved] What is the correct way to implement custom loss function?

[Tofigh_Naghibi (TN)]

It is perhaps too late but for static class you should not make an instance by LSE_loss() instead you should call apply:

loss = LSE_loss.apply