Solved by KFrank in post #2 
              
              
                Hi kliu! 

If I understand your use case, you should start with 
BCEWithLogitsLoss as your loss function (and only change 
to something else if you have a good reason and testing shows 
that the change is for the better). 

Just to be clear, for BCEWithLogitsLoss, the last layer of your 
network sho…