pytorch:NLLLoss ignore_索引默认值

Question

在pytorch NLLLossdoc中，忽略索引的默认值是-100，而不是通常的None，有什么特殊原因吗？似乎任何负值都是等价的

顺便问一下，我想忽略索引的原因可能是什么？谢谢

Answer 1

ignore_index的值必须是int，这就是为什么默认值是int而不是None。默认值是任意的，它可以是任何负数，即任何不是“有效”类标签的值。该函数将忽略目标实例具有该类标签的所有元素。实际上，该选项可用于识别未标记的像素，例如在密集预测任务中

---

Edit：追溯^{}的实现，我们可以在^{}的nll_loss实现中找到以下注释：

# in onnx NegativeLogLikelihoodLoss specification, ignore_index is optional without default value.
# therefore we need to set ignore_index attribute even if it is not specified (e.g. ignore_index=-100).
ignore_index = sym_help._maybe_get_const(ignore_index, "i")