wassdistance

wassdistance copied to clipboard

Added device option to Wasserstein

1

Updated wasserstein submodule to include device

sAbhay

AleksandarPetrov

Thanks a lot for your work. But I don't know how to use this code, it does not like python main.py?

heerduo

Question about using it as a loss function

1

I applied the loss function in layers.py in my training, but I found it doesn't work. When I was trying to train a simple CNN with MNIST dataset, I checked...

Namelessness5

Sinkhorn Iteration

2

Thank you for your work! It gives me a more clear understanding of OT. But I am new to the Sinkhorn iteration algorithm, and as I understand it, the iteration...

zhenyuzhouiu

Very beautiful notebook! However, I believe your implementation of the Sinkhorn distance is faulty: The formal definition of the Sinkhorn Distance (SD) implies that the SD between two translated point...

chrzzl