Solved by KFrank in post #3 
              
              
                Hi Julia! 

In terms of comparing your predictions to the “zeroes and ones” of 
your label, BCEWithLogitsLoss does precisely this (without converting 
your predictions into 1s and 0s). 
BCEWithLogitsLoss takes predictions that are raw-score logits 
(such as those produced by your final Linear layer …