New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable to set parameters for inner Sinkhorn loop in Gromov-Wasserstein #160
Comments
Thanks Dominik for the great comment. I think this is possible using the most recent API, as illustrated in this colab: https://colab.research.google.com/drive/1YWKKjSWnpbhpeMae2tlkHKLLZXP97Cij?usp=sharing You need to instantiate manually a
and then simply make the call
In the colab you can see how the Using such classes, it is easy to have access to detailed granularity on the components that make up an OT problem. I think it's also a lot more viable than overloading the signature of |
Thanks Marco! |
yes, the example I provided is clearly te way Michal and I want to go. We'll still maintain the |
Amazing, thank you! |
Currently, it is not possible to set parameters like
max_iterations
andmin_iterations
of the inner Sinkhorn loop when calling WassersteinSolver or GromovWasserstein because of the same argument names.It would be great to rename the outer loop parameters to have more control about the algorithm.
The text was updated successfully, but these errors were encountered: