Solved by KFrank in post #2 
              
              
                Hi Rafael! 

For a binary classification problem, BCEWithLogitsLoss 
should be your go-to loss function.  (You would only want to 
use BCELoss if your network naturally emits probabilities, 
which it almost certainly doesn’t.) 

Consider carefully the trade-off that you are implying here.  As 
an ex…