search

bubbliiiing / yolox-pytorch

这是一个yolox-pytorch的源码，可以用于训练自己的模型。
Apache License 2.0
888 stars 185 forks source link

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

Open Kakahashi opened 1 year ago

Kakahashi commented 1 year ago

修改了代码，原来的csplayer 替换成swim transformer以后出现一下问题 yolotraining文件该行报错 , 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 commented 1 year ago

同问