EMD module giving error

Open Poulami-Sarkar opened this issue 3 years ago • 1 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

Dec 06 '22 10:12 Poulami-Sarkar

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

Feb 15 '23 04:02 AlphaPav