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

Add MMCR Loss #1446

Merged
merged 5 commits into from
Dec 13, 2023
Merged

Add MMCR Loss #1446

merged 5 commits into from
Dec 13, 2023

Conversation

johnsutor
Copy link
Contributor

This commit addresses #1423, implementing the Maximum Manifold Capacity loss and transforms. The default parameters are set to those used for ImageNet in the original paper https://arxiv.org/pdf/2303.03307.pdf

@johnsutor johnsutor force-pushed the maximum_manifold_capacity branch from d116654 to 5e95fc5 Compare December 10, 2023 20:17
Add MMCR Transform (as subclass of MultiViewTransform)
@johnsutor johnsutor force-pushed the maximum_manifold_capacity branch from 5e95fc5 to a22fd68 Compare December 10, 2023 20:19
@philippmwirth
Copy link
Contributor

Hi @johnsutor, thank you for your contribution! We'll have someone from the team review the pull-request asap.

Copy link
Contributor

@guarin guarin left a comment

Choose a reason for hiding this comment

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

Thanks for the PR! This looks great! I would suggest to expose the BYOLView1Transform arguments in MMCRTransform to make them easily accessible. After that the PR should be ready to merge :)

Copy link

codecov bot commented Dec 12, 2023

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (66ad1b4) 85.24% compared to head (c551512) 85.34%.

Additional details and impacted files
@@            Coverage Diff             @@
##           master    #1446      +/-   ##
==========================================
+ Coverage   85.24%   85.34%   +0.09%     
==========================================
  Files         130      132       +2     
  Lines        5517     5547      +30     
==========================================
+ Hits         4703     4734      +31     
+ Misses        814      813       -1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Copy link
Contributor

@guarin guarin left a comment

Choose a reason for hiding this comment

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

LGTM! Thanks for the updates!

@guarin guarin merged commit f3fd4a3 into lightly-ai:master Dec 13, 2023
6 checks passed
This was referenced Dec 13, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants