Hey Bram,  We’re not comparing loss between two runs. We’re calculating loss twice using the same logit scores. In the same step at exactly the same time. So, Can you check the code, let me know where’s the issue arises.