NLLLoss — PyTorch 1.10.1 documentation
pytorch.org › docs › stableNLLLoss. class torch.nn.NLLLoss(weight=None, size_average=None, ignore_index=- 100, reduce=None, reduction='mean') [source] The negative log likelihood loss. It is useful to train a classification problem with C classes. If provided, the optional argument weight should be a 1D Tensor assigning weight to each of the classes.
BCEWithLogitsLoss — PyTorch 1.10.1 documentation
pytorch.org › docs › stableBCEWithLogitsLoss. class torch.nn.BCEWithLogitsLoss(weight=None, size_average=None, reduce=None, reduction='mean', pos_weight=None) [source] This loss combines a Sigmoid layer and the BCELoss in one single class. This version is more numerically stable than using a plain Sigmoid followed by a BCELoss as, by combining the operations into one ...