Hippogriff / parsenet-codebase

Code base of ParSeNet: ECCV 2020.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Issue on running test.py

guohaoxiang opened this issue · comments

When I download all the dataset and pretrained models and run 'python test.py 0 3999', the following error occur:

Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)

Could you please give me some suggestions on how to fix this?

Can you try single gpu instead of multi gpu? It seems tensors in some operation are on different gpu devices.

It works now! Thanks a lot for your suggestion. If I want to forward my own point cloud data (already stored in h5 format) to ParSeNet to get the segmentation label, which command should I use?

hello, when i run "bash download_dataset.sh" according to readme, i can't found the file "predictions.h5", I also can't download the file on the website. I'm guessing you might still have that dataset, and if so, can you send me a copy, thank!