Pytorch之CrossEntropyLoss() 与 NLLLoss() 的区别

  • NLLLoss 的 输入 是一个对数概率向量和一个目标标签(不需要是one-hot编码形式的). 它不会为我们计算对数概率. 适合网络的最后一层是log_softmax. 损失函数 nn.CrossEntropyLoss() 与 NLLLoss() 相同, 唯一的不同是它为我们去做 softmax.

原文地址:https://www.cnblogs.com/ranjiewen/p/10059490.html