最近在使用MMdetection训练数据,在使用reppionts方法训练时候,出现了错误:RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
首先,排除其他的错误。其他模型训练都没有错误,查找报错位置:
报错在92行,同时86行是没有显示设备的所以可能是这里的问题:
points_range = torch.arange(points.shape[0])
所以将出错的位置修改到同一个设备上:
points_range = torch.arange(points.shape[0]).to(gt_bboxes_lvl.device)
正常训练了。