Solved by ptrblck in post #2 
              
              
                The docs will give you some information about these loss functions as well as small code snippets. 
For a binary classification, you could either use nn.BCE(WithLogits)Loss and a single output unit or nn.CrossEntropyLoss and two outputs. 
Usually nn.CrossEntropyLoss is used for a multi-class classif…