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)
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() ??