用于多分类问题作为损失函数,使用的时候要注意它包含了logsoftmax和nllloss的过程,输入其中的预测值不需要经过softmax,另外输入的真实标签值应当是长整型。
关于torch.nn.CrossEntropyLoss
最新推荐文章于 2024-07-17 13:27:19 发布
用于多分类问题作为损失函数,使用的时候要注意它包含了logsoftmax和nllloss的过程,输入其中的预测值不需要经过softmax,另外输入的真实标签值应当是长整型。