Skip to content
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

tracking progress of gromov-wasserstein #351

Merged
merged 4 commits into from
Apr 24, 2023

Conversation

bosr
Copy link
Contributor

@bosr bosr commented Apr 7, 2023

Yet another progress bar PR ;)

  • Add support and doc for reporting progress in the GromovWasserstein solver
  • Expose GWState in the docs: [EDIT: removed] this actually causes linting issues because it is already exposed via autosummary
  • Rework the notebook tutorial to show how to track progress for Sinkhorn, LRSinkhorn and GromovWasserstein

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@codecov-commenter
Copy link

codecov-commenter commented Apr 7, 2023

Codecov Report

Merging #351 (e4a8737) into main (3f78882) will decrease coverage by 0.01%.
The diff coverage is 80.00%.

📣 This organization is not using Codecov’s GitHub App Integration. We recommend you install it so Codecov can continue to function properly for your repositories. Learn more

Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##             main     #351      +/-   ##
==========================================
- Coverage   88.21%   88.20%   -0.01%     
==========================================
  Files          51       51              
  Lines        5481     5496      +15     
  Branches      827      829       +2     
==========================================
+ Hits         4835     4848      +13     
- Misses        527      528       +1     
- Partials      119      120       +1     
Impacted Files Coverage Δ
src/ott/utils.py 66.66% <0.00%> (ø)
src/ott/solvers/quadratic/gromov_wasserstein.py 87.94% <77.77%> (-0.78%) ⬇️
src/ott/solvers/linear/sinkhorn.py 96.83% <100.00%> (ø)
src/ott/solvers/linear/sinkhorn_lr.py 96.28% <100.00%> (+0.11%) ⬆️

@bosr
Copy link
Contributor Author

bosr commented Apr 7, 2023

I'll add to GromovWasserstein the same kind of test that I had added to Sinkhorn.

@michalk8 michalk8 added the enhancement New feature or request label Apr 7, 2023
Copy link
Collaborator

@michalk8 michalk8 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks a lot @bosr , got only very minor comments.
For the tests, we can add this later and am ok merging without.

One question: have you tried passing the callback to GW, as well as it's linear solver? Might be nice to show in the notebook.

src/ott/solvers/quadratic/gromov_wasserstein.py Outdated Show resolved Hide resolved
@marcocuturi
Copy link
Contributor

what's the situation on this ? @bosr / @michalk8

@michalk8 michalk8 merged commit cd8bd4e into ott-jax:main Apr 24, 2023
@michalk8
Copy link
Collaborator

Thanks @bosr for the contribution - I just fixed some minor issues.

michalk8 added a commit that referenced this pull request Jun 27, 2024
* tracking progress of gromov-wasserstein

* fix: linting issue

* fix: do not expose GWState explicitly

* Update notebook, fix minor issues

---------

Co-authored-by: Michal Klein <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants