-
Notifications
You must be signed in to change notification settings - Fork 78
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
Feature/ Introduce initialization methods for Sinkhorn #98
Conversation
@marcocuturi could you please enable the readthedocs build? |
…oduce defaultinit class
@michalk8 : i had some issues with the webhook to enable recompiling docs on PR. I will look into this soon. |
@JTT94 could you please fix the bibtex file and deal with the rest of the comments? After that, think we can merge. |
Thanks for the feedback, this is good to be merged |
Also just noticed that the initializers are not registered as PyTrees, i.e.
fails. Could you please register them using |
Thanks a lot @JTT94 , could you please include the last change mentioned here: #98 (comment) ? It's more efficient than doing |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @JTT94, LGTM, merging this!
Great! And thanks for your patience with feedback, I appreciate it |
Examples here: https://colab.research.google.com/drive/1vncmDEr3t6_OKfVC0PJin8PIRBViPyO0?usp=sharing
Initialization methods stored in /core/initializers.py
SinkhornInitializer
init_dual_a
andinit_dual_b
lse_mode
, also handling entries for 0 weightsModification to Sinkhorn api,
init_dual_a
/init_dual_b
vectors are passed to solver