@urw7rs I have used BCEWithLogitsLoss, which comes built-in with sigmoid activation function. So it’s behaviour should stay the same.
Besides, I also have tried replacing the above loss function with nn. BCELoss and applying sigmoid activation function at last layer, as exactly what you said, but still loss was not converging.