AttributeError: module 'nearest_neighbors' has no attribute 'knn_batch'

tsunghan-wu / RandLA-Net-pytorch

:four_leaf_clover: Pytorch Implementation of RandLA-Net (https://arxiv.org/abs/1911.11236)

MIT License

122 stars 33 forks source link

AttributeError: module 'nearest_neighbors' has no attribute 'knn_batch' #29

Open Xavier-wa opened 1 year ago

Xavier-wa commented 1 year ago

I try to train it by the way of ReadMe.md,But there are some problems: File "/home/xavier/RidarSS/RandLA-Net-pytorch-master/helper_tool.py", line 171, in knn_search neighbor_idx = nearest_neighbors._knn_batch(support_pts, query_pts, k, omp=True) AttributeError: module 'nearest_neighbors' has no attribute 'knn_batch' any solution?plz