-
Notifications
You must be signed in to change notification settings - Fork 290
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
Add MMCR Loss #1446
Conversation
d116654
to
5e95fc5
Compare
Add MMCR Transform (as subclass of MultiViewTransform)
5e95fc5
to
a22fd68
Compare
Hi @johnsutor, thank you for your contribution! We'll have someone from the team review the pull-request asap. |
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 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 :)
Co-authored-by: guarin <[email protected]>
Co-authored-by: guarin <[email protected]>
… from paper) Format code and fix imports in MMCR transform file
Codecov ReportAll modified and coverable lines are covered by tests ✅
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. |
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.
LGTM! Thanks for the updates!
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