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

ImportError: cannot import name 'sparse_quantize #80

Closed knifeven closed 2 years ago

knifeven commented 2 years ago

Hi, I tested spvnas with SemanticKITTI I got the latest version of spvnas and torchsparse (master) I showed the following error:

Traceback (most recent call last): File "visualize.py", line 14, in from torchsparse.utils import sparse_quantize ImportError: cannot import name 'sparse_quantize

my env: python 3.6 cuda 10.1 torch 1.6.0 torchvision 0.7.0 torchpack 0.3.1

How can I solve this?

OmerCahana commented 2 years ago

change to : from torchsparse.utils.quantize import sparse_quantize

zhijian-liu commented 2 years ago

Thanks @OmerCahana! Would you mind submitting a PR to fix this issue?