File "D:\pythonProject\venv\lib\site-packages\torch\nn\functional.py", line 2385, in nll_loss ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index) IndexError: Target 3 is out of bounds.
时间: 2023-09-07 20:13:11 浏览: 155
Python库 | venv_kernel-1.0.0-py3-none-any.whl
这个错误是由于目标索引超出了范围,通常可能是由于目标标签中包含了超出了模型输出数量的值。您可以检查一下您的目标标签是否正确,并与模型输出数量匹配。如果您使用的是 PyTorch,还可以考虑使用 CrossEntropyLoss 代替 nll_loss 函数来计算损失,它会自动将目标标签转换为 one-hot 编码形式。
阅读全文