microsoft / SpareNet

Style-based Point Generator with Adversarial Rendering for Point Cloud Completion (CVPR 2021)

Home Page:https://arxiv.org/abs/2103.02535

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

EMD module giving error

Poulami-Sarkar opened this issue · comments

When I run the inference script with the following command , python3 test.py --gpu 0 --workdir ./ --model sparenet --weights SpareNet.pth --test_mode defaul
I get this error:

emd.forward( AttributeError: module 'emd' has no attribute 'forward' (torchenv) sarkar@devcube:~/Github/SpareNet$

My environment is as follows:
CUDA Version: 11.6
Python 3.8.10
PyTorch 1.13.0+cu116

Thanks for the question! Did you install the cuda packages? sh setup_env.sh