PU-Net_pytorch
PU-Net_pytorch copied to clipboard
The emd_loss has problem for backpropagation
I tried to use "get_emd_loss.py" but I met this problem
RuntimeError: Expected isFloatingType(grads[i].type().scalarType()) to be true, but got false. (Could this error message be improved? If so, please report an enhancement request to PyTorch.)
Could you resolve it?