BCEWithLogitsLoss() and sigmoid layer

No, using threshold 0 on your raw output is equivalent to the threshold of 0.5 after a sigmoid layer.

1 Like