Ah ok cool.
So i’m under the assumption that softmax is automatically computed in the nn.CrossEntropyLoss module and nn.BCE(WithLogits)Loss will also have sigmoid computed within, whereas for nn.BCELoss you need to apply sigmoid first? And if so, I can send the output of sigmoid + nn.BCELoss to nn.CrossEntropyLoss?