Solved by ptrblck in post #4 
              
              
                You could deal with a binary classification use case in different ways: 

You could use a single output and treat the output as the logit (or probability) representing the nagative and positive class. For this use case you would use nn.BCEWithLogitsLoss (or nn.BCELoss, if you are applying a sigmoid …