Pytorch学习笔记——NLLLoss & CrossEntropyLoss

NLLLoss(Negative Log Likelihood Loss)图解如下:

Pytorch学习笔记——NLLLoss & CrossEntropyLoss_第1张图片

CrossEntropyLoss(交叉熵损失函数)图解如下:

Pytorch学习笔记——NLLLoss & CrossEntropyLoss_第2张图片

二者关系为:

LogSoftmax+ NLLLoss <===> CrossEntropyLoss

官方文档:

        CrossEntropyLoss — PyTorch 1.10 documentation

        NLLLoss — PyTorch 1.10 documentation

你可能感兴趣的:(Deep,Learning,pytorch,学习,人工智能)