I had the problem too, but the inspirations above helped me.
In the end, I forgot to pass the Sigmoid function (or whatever function you use to get the probability)
prob_predict = torch.nn.Sigmoid()(logits_predict)