search

Na-Z / attMPTI

[CVPR 2021] Few-shot 3D Point Cloud Semantic Segmentation
MIT License
166 stars 31 forks source link

Update mpti.py. There are some wrong usage about F.pairwise_distance #25

Closed AaronNZH closed 1 year ago

AaronNZH commented 2 years ago

There are some wrong usage about F.pairwise_distance, which lead to error data shape, and then generate nan data. Finaly, the nan data will make faiss.IndexFlatL2 produce -1 in array I, which means array out of bound error in torch.gather