search

Julie-tang00 / Point-BERT

[CVPR 2022] Pre-Training 3D Point Cloud Transformers with Masked Point Modeling
MIT License
541 stars 65 forks source link

EMD Extension: Error while attempting to load the dVAE model #69

Open vijay-jaisankar opened 1 year ago

vijay-jaisankar commented 1 year ago

Hi, thanks for open-sourcing your work! It seems very interesting and I'd love to try it out, but I'm facing an issue with the same.

Context

I managed to install the dependencies as mentioned in the README. The emd extension was not installing at first, so I made this modification.

Issue

However, while trying to create an instance of the dVAE model, the following issue cropped up:

from models.dvae import DCGNN

resulted in an ImportError: 

ImportError: /root/.local/lib/python3.10/site-packages/emd_ext-0.0.0-py3.10-linux-x86_64.egg/emd_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _Z16MatchCostForwardN2at6TensorES0_S0_

Here is the full stack trace: ImportError: Entire stack trace "ImportError: Entire stack trace"

Thanks

LilRedWu commented 1 year ago

Hi, I also face this issue that the emd can't be installed successfully. Have you solved this problem?

vijay-jaisankar commented 1 year ago

Hi @LilRedWu , unfortunately I've not been able to solve this yet.

The only solution seems to be using another implementation of EMD, which I am trying out.

yuxumin commented 1 year ago

Hi, In fact EMD loss is not used in our dVAE. comment out the line could solve the problem.