indices should be either on cpu or on the same device as the indexed tensor (cpu)

[uwuwn]

[YJJ0829]

I came with the same error too, have you solved it already?

[xiaoxiae]

Got the same issue.

[LDP626]

@xiaoxiae @YJJ0829 Add a line proposals_idx = proposals_idx.to(mask_inds.device) before the line cur_proposals_idx = proposals_idx[mask_inds].long() in the softgroup/model/softgroup.py file.