hi , i have this problem too.My model can’t learn during the training .I changed Dice loss function for many times but nothing changed.
but i didn’t use custom backward function,i just used forward function. Have you used it ? Was it useful?