深度阅读

torch KLDivLoss计算损失出现负数

作者
作者
2023年08月22日
更新时间
2.3 分钟
阅读时间
0
阅读量

KLDivLoss损失出现负数

  1. KL散度的input部分需要经过log的计算
  2. 分布结果相加需要为1

用于连续分布的距离度量;并且对离散采用的连续输出空间分布进行回归通常很有用;用label_smoothing就采用这个;另外知识蒸馏更是需要依赖它。

官方文档

https://pytorch.org/docs/stable/generated/torch.nn.KLDivLoss.html

博客作者

热爱技术,乐于分享,持续学习。专注于Web开发、系统架构设计和人工智能领域。