zou-longkun/GAST

emd_cuda

Jocelynan19 opened this issue · 1 comments

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'