RuntimeError: invalid argument 5: k not in range for dimension at C:/w/b/windows/pytorch/aten/src\THC/generic/THCTensorTopK.cu:24

[Kakahashi]

修改了代码，原来的csplayer 替换成swim transformer以后出现一下问题
yolo_training文件该行报错 _, pos_idx = torch.topk(cost[gt_idx], k=dynamic_ks[gt_idx].item(), largest=False)
RuntimeError: invalid argument 5: k not in range for dimension at C:/w/b/windows/pytorch/aten/src\THC/generic/THCTensorTopK.cu:24
Epoch 1/100: 0%| | 0/253 [00:12<?, ?it/s<class 'dict'>]
请问一下什么原因，如何解决啊（另外后来无意中把刚开始的focus切片操作修改以后，发现能正常训练，但是这个修改可能影响实验，所以寻求其他的解决办法）

[XinyueZhangqdu]

同问