Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

python demo.py pv_rcnn.yaml error occurs,however pointpillar.yaml、pointrcnn.yaml ,OK ,why #3

Open
wangwei-gpu opened this issue Nov 20, 2020 · 0 comments

Comments

@wangwei-gpu
Copy link

python demo.py --cfg_file cfgs/kitti_models/pv_rcnn.yaml --ckpt pth/pv_rcnn_8369.pth --data_path ../testing/velodyne/000006.bin 。The following error occurs:
INFO Visualized sample index: 1
Traceback (most recent call last):
File "demo.py", line 103, in
main()
File "demo.py", line 91, in main
pred_dicts, _ = model.forward(data_dict)
File "/home/dfmc/OpenPCDet/pcdet/models/detectors/pv_rcnn.py", line 11, in forward
batch_dict = cur_module(batch_dict)
File "/home/dfmc/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(*input, **kwargs)
File "/home/dfmc/OpenPCDet/pcdet/models/backbones_3d/spconv_backbone.py", line 139, in forward
x = self.conv_input(input_sp_tensor)
File "/home/dfmc/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(*input, **kwargs)
File "/home/dfmc/.local/lib/python3.6/site-packages/spconv/modules.py", line 134, in forward
input = module(input)
File "/home/dfmc/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(*input, **kwargs)
File "/home/dfmc/.local/lib/python3.6/site-packages/spconv/conv.py", line 181, in forward
use_hash=self.use_hash)
File "/home/dfmc/.local/lib/python3.6/site-packages/spconv/ops.py", line 95, in get_indice_pairs
int(use_hash))
ValueError: /home/dfmc/spconv-master/src/spconv/spconv_ops.cc 87
unknown device type,What is the problem? thank you very much!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant