MMdetection报错不在同一个设备上

最近在使用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)

正常训练了。