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

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

Open iooops opened this issue 2 years ago • 5 comments

跑训练的时候报错

Dec 21 '22 10:12 iooops

candidate_starts = candidate_starts.view(-1)[flattened_candidate_mask]    # [num_candidates]

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

Dec 21 '22 10:12 iooops

遇到了同样的问题，请问你解决了吗，是怎么解决的？

Apr 09 '23 01:04 cuikai-ai

没有

Apr 10 '23 17:04 iooops

也遇到了这个gpu和cpu乱用的问题，求解

Jun 02 '23 03:06 lixiaoxiaobin

出问题的代码添加to("cuda:0")

Dec 18 '23 08:12 heguyu