损失函数NLLLoss() 的 输入 是一个对数概率向量和一个目标标签. 它不会为我们计算对数概率,适合最后一层是log_softmax()的网络. 损失函数 CrossEntropyLoss() 与 NLLLoss() 类似, 唯一的不同是它为我们去做 log_softmax.可以理解为:
CrossEntropyLoss=nll_loss(log_softmax(input, 1), target)
损失函数NLLLoss() 的 输入 是一个对数概率向量和一个目标标签. 它不会为我们计算对数概率,适合最后一层是log_softmax()的网络. 损失函数 CrossEntropyLoss() 与 NLLLoss() 类似, 唯一的不同是它为我们去做 log_softmax.可以理解为:
CrossEntropyLoss=nll_loss(log_softmax(input, 1), target)
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。