search

Vegeta2020 / SE-SSD

SE-SSD: Self-Ensembling Single-Stage Object Detector From Point Cloud, CVPR 2021.
Apache License 2.0
811 stars 128 forks source link

RuntimeError: Expected object of backend CUDA but got backend CPU for argument #2 'mat2' #93

Open Parkkkkk opened 1 year ago

Parkkkkk commented 1 year ago

Traceback (most recent call last): File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/tools/train.py", line 120, in main() File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/tools/train.py", line 117, in main train_detector(model, datasets, cfg, distributed=distributed, validate=args.validate, logger=logger) File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/det3d/torchie/apis/train_sessd.py", line 329, in train_detector trainer.run(data_loaders, cfg.workflow, cfg.total_epochs, local_rank=cfg.local_rank) File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/det3d/torchie/trainer/trainer_sessd.py", line 472, in run epoch_runner(data_loaders[0], data_loaders[1], self.epoch, kwargs) File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/det3d/torchie/trainer/trainer_sessd.py", line 346, in train outputs = self.batch_processor_inline(self.model, self.model_ema, data_batch, consistency_weight, train_mode=True, kwargs) File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/det3d/torchie/trainer/trainer_sessd.py", line 265, in batch_processor_inline output_ema = model_ema(example, is_ema=[True, None]) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call result = self.forward(*input, kwargs) File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/det3d/models/detectors/voxelnet_sessd.py", line 34, in forward x = self.extract_feat(data) File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/det3d/models/detectors/voxelnet_sessd.py", line 13, in extract_feat x = self.backbone(input_features, data["coors"], data["batch_size"], data["input_shape"]) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call result = self.forward(*input, *kwargs) File "/home/hyeokjin_1804/Desktop/workspace/SE-SSD/det3d/models/backbones/scn.py", line 183, in forward ret = self.middle_conv(ret) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call result = self.forward(input, kwargs) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/spconv/modules.py", line 125, in forward input = module(input) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call result = self.forward(*input, **kwargs) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/spconv/conv.py", line 161, in forward outids.shape[0]) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/spconv/functional.py", line 83, in forward return ops.indice_conv(features, filters, indice_pairs, indice_pair_num, num_activate_out, False, True) File "/home/hyeokjin_1804/anaconda3/envs/sesd/lib/python3.6/site-packages/spconv/ops.py", line 114, in indice_conv int(inverse), int(subm)) RuntimeError: Expected object of backend CUDA but got backend CPU for argument #2 'mat2' (checked_tensor_unwrap at /opt/conda/conda-bld/pytorch_1556653099582/work/aten/src/ATen/Utils.h:70)

where should i put .cuda() ??