[debug] RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘

最新推荐文章于 2024-04-11 09:31:37 发布

Harry嗷

最新推荐文章于 2024-04-11 09:31:37 发布

阅读量1.2w

点赞数 12

分类专栏： BUG解决 Python 文章标签： python pycharm pytorch

本文链接：https://blog.youkuaiyun.com/qq_41683065/article/details/122441927

版权

Python 同时被 2 个专栏收录

62 篇文章

订阅专栏

BUG解决

27 篇文章

订阅专栏

本文描述了在使用PyTorch的nn.CrossEntropyLoss时遇到的一个运行时错误，并提供了详细的错误堆栈跟踪。错误源于数据类型不匹配，文章进一步提供了解决方案——确保目标张量的数据类型为torch.int64。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

在使用nn.CrossEntropyLoss时出现报错：

loss_id = self.IDLoss(id_preds, id_targets)

Traceback (most recent call last):
  File "/home/hgx/pycharm-2020.1.3/plugins/python/helpers/pydev/_pydevd_bundle/pydevd_exec2.py", line 3, in Exec
    exec(exp, global_vars, local_vars)
  File "<input>", line 1, in <module>
  File "/home/hgx/anaconda3/envs/bytetrack/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home/hgx/anaconda3/envs/bytetrack/lib/python3.7/site-packages/torch/nn/modules/loss.py", line 1152, in forward
    label_smoothing=self.label_smoothing)
  File "/home/hgx/anaconda3/envs/bytetrack/lib/python3.7/site-packages/torch/nn/functional.py", line 2846, in cross_entropy
    return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'

问题分析

数据类型导致的错误

解决方案

进行数据类型转换

id_targets = id_targets.to(torch.int64)     # TODO: ReID. [matched_anchor], float16 to int