PyTorch交叉熵损失函数详解

注意点：

  1. torch.nn.CrossEntropyLoss中的ignore index 是作用在target上
  2. torch.nn.CrossEntropyLoss 最终交叉熵的值是平均值
  3. 统计平均值的时候，不会统计ignore index的
  4. 公式中的值是以自然数e为底
  5. 对于 C 类分类任务，交叉熵损失的初始值通常在 log(C) 附近 log(1w) = 9.2 e为底
posted @ 2025-03-24 21:14  xiezhengcai  阅读(159)  评论(0)    收藏  举报