version of torchsparse and python - Githubissues

mit-han-lab / spvnas

[ECCV 2020] Searching Efficient 3D Architectures with Sparse Point-Voxel Convolution

http://spvnas.mit.edu/

MIT License

587 stars 109 forks source link

version of torchsparse and python #77

Closed ginobilinie closed 3 years ago

ginobilinie commented 3 years ago

Hi,

I have followed the instructions in this repo to install all the dependencies, but when I try to run some experiments, I found the error below:

File "/home/xxx/codebooks/spvnas/core/datasets/semantic_kitti.py", line 7, in from torchsparse.utils import sparse_collate_fn, sparse_quantize ImportError: cannot import name 'sparse_collate_fn'

These are my environments: OS: ubuntu 18.04 python: 3.6 pytorch: 1.6.0 cuda: 10.2 torchsparse: 1.4.0

Thanks.

zhijian-liu commented 3 years ago

Could you pull the latest SPVNAS codebase instead?

ginobilinie commented 3 years ago

Thank you so much. Let me try it.