To change to BCEWithLogitsLoss, would I simply get rid of my final Sigmoid layer, and during training, pass those outputs to the loss function? In my validation, I’ll have to call the sigmoid function on my outputs before i compare predicted labels and ground truths, right? Am I missing anything?