关于yolox_head_fast.py里面dynamic_k_matching函数的一个小bug

[ahjcahkl]

在参考作者代码对官方pipeline进行修改之后，终于统一了数据格式的问题，但是后来又遇到了一个问题，与合理的gt数目有关，源代码是这样的：

# 如果有预测框（花心大萝卜）匹配到了1个以上的gt时，做特殊处理。
       if (anchor_matching_gt > 1).float().sum() > 0:

           # 首先，找到与花心大萝卜具有最小cost的gt。

            # 找到 花心大萝卜 的下标（这是在anchor_matching_gt.shape[N, A]中的下标）。假设有R个花心大萝卜。

           indexes = torch.where(anchor_matching_gt > 1)

           index = torch.stack((indexes[0], indexes[1]), 1)  # [R, 2]  每个花心大萝卜2个坐标。第0个坐标表示第几张图片，第1个坐标表示第几个格子。

我的代码会在进入这个判断后稳定跑飞，报错位置在求取下标index这一步

indexes = torch.where(anchor_matching_gt > 1)

报错信息为：

RuntimeError: numel: integer multiplication overflow

不太清楚这个where计算量在哪里，感觉非常的迷惑，希望能指点一下为什么会出现这个问题