EMD module giving error
Poulami-Sarkar opened this issue · comments
Poulami Sarkar commented
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
AlphaPav commented
Thanks for the question! Did you install the cuda packages? sh setup_env.sh