ERROR 使用pytorch的函数 torch.nn.CrossEntropyLoss()计算Loss时报错 或者 loss = criterion(output, target) 报错： RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed 解决方法： 原因一：模型输出与分类数不一致 看模型的输出尺寸与分类数差异是否明显，核查代码是否存在错误。 如果没有错误，只是映射维度不对，可以考虑在模型的最后一层加一层FC层，将输出尺寸
超级会员免费看
在使用Pytorch的CrossEntropyLoss计算损失时遇到错误。问题可能出在模型输出与分类数不匹配或标签未从0开始。检查模型输出维度是否与类别数一致，若不一致需调整模型或添加全连接层。另外，确保标签值是从0开始的，否则会导致该错误。
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
