pytorch CrossEntropyLoss 的数值稳定性问题

手写了一个带 maskCrossEntropyLoss, 分为了 softmax, lognllloss 三个步骤。
先 softmax, 再log + nllloss出现inf, 先 log_softmax, 再nllloss就可以了。

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页