Error 'RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) ' in line 147 of train_retrieval.py

[Qusijia]

Thank you for your work. May I ask how I can make the necessary changes if you encounter the above issue? Do you have any suggestions you can give me?

[jucic]

topk_idx->topk_idx.cpu()