src/transformers/models/deta/modeling_deta.py/DetaStage2Assigner/postprocess_indices
class DetaStage2Assigner:    def postprocess_indices(self, pr_inds, gt_inds, iou):
        return sample_topk_per_gt(pr_inds, gt_inds, iou, self.k)