RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) when run demo.py in colab

[Pathomphop]

/usr/local/lib/python3.10/dist-packages/torch/utils/_contextlib.py:125: UserWarning: Decorating classes is deprecated and will be disabled in future versions. You should only decorate functions or methods. To preserve the current behavior of class decoration, you can directly decorate the __init__ method and nothing else. warnings.warn("Decorating classes is deprecated and will be disabled in " [05/17 07:22:07 detectron2]: Arguments: Namespace(config_file='configs/BAText/ICDAR2015/v1_attn_R_50.yaml', webcam=False, video_input=None, input=['/content/images/'], output='output', confidence_threshold=0.3, opts=['MODEL.WEIGHTS', 'v1_ic15_finetuned.pth']) WARNING [05/17 07:22:07 d2.config.compat]: Config 'configs/BAText/ICDAR2015/v1_attn_R_50.yaml' has no VERSION. Assuming it to be compatible with latest v2. [05/17 07:22:08 d2.checkpoint.detection_checkpoint]: [DetectionCheckpointer] Loading from v1_ic15_finetuned.pth ... The checkpoint state_dict contains keys that are not used by the model: pixel_mean pixel_std 0% 0/10 [00:00<?, ?it/s]/usr/local/lib/python3.10/dist-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.) return _VF.meshgrid(tensors, *kwargs) # type: ignore[attr-defined] 0% 0/10 [00:01<?, ?it/s] Traceback (most recent call last): File "/content/AdelaiDet/demo/demo.py", line 87, in predictions, visualized_output = demo.run_on_image(img) File "/content/AdelaiDet/demo/predictor.py", line 54, in run_on_image predictions = self.predictor(image) File "/content/detectron2/detectron2/engine/defaults.py", line 317, in call predictions = self.model([inputs])[0] File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(args, **kwargs) File "/content/AdelaiDet/adet/modeling/one_stage_detector.py", line 99, in forward return self.inference(batched_inputs) File "/content/AdelaiDet/adet/modeling/one_stage_detector.py", line 169, in inference return OneStageRCNN._postprocess(results, batched_inputs, images.image_sizes) File "/content/AdelaiDet/adet/modeling/one_stage_detector.py", line 185, in _postprocess r = detector_postprocess(results_per_image, height, width) File "/content/AdelaiDet/adet/modeling/one_stage_detector.py", line 16, in detector_postprocess results = d2_postprocesss(results, output_height, output_width, mask_threshold) File "/content/detectron2/detectron2/modeling/postprocessing.py", line 58, in detector_postprocess results = results[output_boxes.nonempty()] File "/content/detectron2/detectron2/structures/instances.py", line 142, in getitem ret.set(k, v[item]) RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

[Gorgerbin]

I got the same problem. Do you have solutions?

[Pathomphop]

if you run in colab chnage the version of condacolab