TNX, I also have my own loss function, but using it, my model parameters are not updated and stay the same. I opened another issue; Could you help me with that? I’m struggling with that for about a week and not making any progress.
This is the page of my issue