https://lowcostwallchargeripod.blogspot.com/2022/07/custom-loss-function-implementation.html