in code  but in your paper, the loss is 