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

[leily578]

This is happening when I test Oriented RepPoints. My dataset is DOTAv1 and it could normal run in rotated_faster_rcnn.
Traceback (most recent call last):
File "C:\Code\mmrotate\tools\train.py", line 194, in
main()
File "C:\Code\mmrotate\tools\train.py", line 183, in main
train_detector(
File "c:\code\mmrotate\mmrotate\apis\train.py", line 144, in train_detector
runner.run(data_loaders, cfg.workflow)
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmcv\runner\epoch_based_runner.py", line 136, in run
epoch_runner(data_loaders[i], **kwargs)
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmcv\runner\epoch_based_runner.py", line 53, in train
self.run_iter(data_batch, train_mode=True, **kwargs)
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmcv\runner\epoch_based_runner.py", line 31, in run_iter
outputs = self.model.train_step(data_batch, self.optimizer,
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmcv\parallel\data_parallel.py", line 77, in train_step
return self.module.train_step(*inputs[0], **kwargs[0])
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmdet\models\detectors\base.py", line 248, in train_step
losses = self(**data)
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmcv\runner\fp16_utils.py", line 119, in new_func
return old_func(*args, **kwargs)
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmdet\models\detectors\base.py", line 172, in forward
return self.forward_train(img, img_metas, **kwargs)
File "c:\code\mmrotate\mmrotate\models\detectors\single_stage.py", line 81, in forward_train
losses = self.bbox_head.forward_train(x, img_metas, gt_bboxes,
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmdet\models\dense_heads\base_dense_head.py", line 335, in forward_train
losses = self.loss(*loss_inputs, gt_bboxes_ignore=gt_bboxes_ignore)
File "c:\code\mmrotate\mmrotate\models\dense_heads\oriented_reppoints_head.py", line 874, in loss
cls_reg_targets_init = self.get_targets(
File "c:\code\mmrotate\mmrotate\models\dense_heads\oriented_reppoints_head.py", line 798, in get_targets
sampling_result) = multi_apply(
File "C:\Anaconda3\envs\mmrotate\lib\site-packages\mmdet\core\utils\misc.py", line 30, in multi_apply
return tuple(map(list, zip(*map_results)))
File "c:\code\mmrotate\mmrotate\models\dense_heads\oriented_reppoints_head.py", line 675, in _point_target_single
assign_result = assigner.assign(proposals, gt_bboxes, overlaps,
File "c:\code\mmrotate\mmrotate\core\bbox\assigners\convex_assigner.py", line 125, 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)