https://discountedgerberprotectplus.blogspot.com/2024/11/custom-loss-function-implementation.html