Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Mismatch against another EMD library

See original GitHub issue

Excuse my poor understanding towards EMD. I test geomloss with the following code:

points1 = torch.rand(2, 1024, 2, requires_grad=True).cuda()
points2 = torch.rand(2, 1024, 2).cuda()
emd_sinkhorn = geomloss.SamplesLoss(loss='sinkhorn', p=2, blur=0.01, backend='auto')(points1, points2)

Also, I try to use another EMD library (https://github.com/Colin97/MSN-Point-Cloud-Completion/tree/master/emd) which uses auction algorithm:

emd-auction, assignment = emd(points1, points2, eps=0.05, iters=2000)

However, two experiments give different results. How can I change the usage of geomloss to match another one? Thanks for your help very much!

Issue Analytics

State:
Created 3 years ago
Comments:10 (2 by maintainers)

Top GitHub Comments

2reactions

zhangmozhecommented, Jun 30, 2020

Please try debias=False and p=1.

It works when I use the following code:

geomloss.SamplesLoss(loss='sinkhorn', debias=False, p=1, blur=1e-3, scaling=0.999, backend='auto')(points1, points2)

1reaction

wzm2256commented, Jul 1, 2020

😃 You are welcome. This setting let you use entropic regularized (debias=False) EMD (p=1) loss, which is in accordant with the description in the library you mentioned.