search

jeanfeydy / geomloss

Geometric loss functions between point clouds, images and volumes
MIT License
570 stars 57 forks source link

name 'generic_logsumexp' is not defined #52

Closed FieeFlip closed 2 years ago

FieeFlip commented 2 years ago

name 'generic_logsumexp' is not defined

a = torch.rand(28996, 2)
b = torch.rand(28996, 2)

samplesloss = SamplesLoss('sinkhorn')
samplesloss(a, b)
FieeFlip commented 2 years ago

!pip install pykeops is not useful

FieeFlip commented 2 years ago

!pip install geomloss[full] must in google Coblab, it is not useful in Kaggle

FieeFlip commented 2 years ago

To install GeomLoss and all its (optional) dependencies, please: 1 Install the CUDA toolkit, including the nvcc compiler. 2 Install PyTorch. 3 Install the KeOps library. 4 Install GeomLoss with:pip install geomloss

bigian98 commented 2 years ago

Hi FieeFLip, I noticed that this warning message appears to me only when the size of a and b is greater than a certain value; any suggestions?