Focus Detr K out of range error

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

https://detrex.readthedocs.io/en/latest/

Apache License 2.0

2k stars 206 forks source link

Focus Detr K out of range error #354

Open Kunaldargan opened 2 months ago

Kunaldargan commented 2 months ago

detrex/projects/focus_detr/modeling/transformer_layer.py", line 337, in forward select_tgt_index = torch.topk(mc_score, self.topk_sa, dim=1)[1] # bs, nq ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: selected index k out of range

Dear authors, Kindly look into the error reported.

Please explain any solution.