emd_cuda
Jocelynan19 opened this issue · 1 comments
Jocelynan19 commented
i try to run train_Norm.py, but one module not found:
Traceback (most recent call last):
File "/home/cjl/GAST-master/train_Norm.py", line 9, in
from emd import earth_mover_distance
File "/home/cjl/GAST-master/emd.py", line 2, in
import emd_cuda
ModuleNotFoundError: No module named 'emd_cuda'
zou-longkun commented
Please refer to https://github.com/meder411/PyTorch-EMDLoss.