Check logits/sigmoid outputs of each model and wether the losses calc + prediction is consistent with that!