Focus Detr K out of range error

[Kunaldargan]

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.