The training still passes, but the result is not correct comparing to train locally. I think only change the loss is not possible to change the gradient, the gradient is input-related. So until now, I didn’t find a solution for this problem