RuntimeError: multi-target not supported at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:15
时间: 2023-10-03 18:03:38 浏览: 164
apr-util-1.2.2.tar.gz_Linux/Unix编程_Unix_Linux_
这个错误通常表示在使用 PyTorch 的 ClassNLLCriterion 损失函数时,输入的目标值不符合要求。ClassNLLCriterion 要求目标值是一个 LongTensor 类型且每个元素的值必须是 0 到类别数-1 之间的整数。
你可以检查一下你的目标值是否满足要求,并且确保你的模型输出的形状与目标值的形状匹配。如果你仍然无法解决问题,可以尝试使用 PyTorch 的 CrossEntropyLoss 损失函数代替 ClassNLLCriterion。CrossEntropyLoss 将自动执行 softmax 操作并计算交叉熵损失。
阅读全文