使用mmrotate训练sasm模型遇到的BUG：RuntimeError: indices should be either on cpu or on the same device as the

  File "D:\PythonProject\mmrotate-main\mmrotate\core\bbox\assigners\convex_assigner.py", line 127, in assign
    points_index = points_range[lvl_idx]
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

在排除其他错误之后，根据错误提示定位到convex_assigner.py中, line 127

报错在127行，同时120行是没有显示设备的所以可能是这里的问题：

所以将出错的位置修改到同一个设备上：

        # points_range = torch.arange(points.shape[0])
        # 修改到同一个设备上：
        points_range = torch.arange(points.shape[0]).to(gt_bboxes_lvl.device)

就解决了这个问题了。

参考文章：MMdetection报错不在同一个设备上_runtimeerror: indices should be either on cpu or o-优快云博客