search

vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
MIT License
1.38k stars 181 forks source link

Implements SSL-EY #378

Open jameschapman19 opened 7 months ago

jameschapman19 commented 7 months ago

Adding SSL-EY from (https://arxiv.org/abs/2310.01012, https://neurips.cc/virtual/2023/80864)

For the moment I've implemented the simple version but it should in principle plug into multicrop like SWAV.

Will have a stab at running the shorter benchmarks shortly.

vturrisi commented 7 months ago

hey @jameschapman19 thanks for contributing. I'll try to start checking your code soon.

jameschapman19 commented 7 months ago

@vturrisi - just working through ironing things out perhaps a premature PR I just wanted to get a hang of the CI. Will give you a shout when I'm passing more of these.

vturrisi commented 7 months ago

@jameschapman19 no worries, take your time :) and ping me when you are done

codecov[bot] commented 7 months ago

Codecov Report

Merging #378 (9e8ad7e) into main (0eadf3d) will increase coverage by 0.12%. Report is 1 commits behind head on main. The diff coverage is 93.75%.

:exclamation: Current head 9e8ad7e differs from pull request most recent head 11eac92. Consider uploading reports for the commit 11eac92 to get more accurate results

Additional details and impacted files | [Flag](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378/flags?src=pr&el=flags&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) | Coverage Δ | | *Carryforward flag | |---|---|---|---| | [cpu](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) | `80.02% <93.75%> (+0.15%)` | :arrow_up: | | | [dali](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) | `38.81% <ø> (ø)` | | Carriedforward from [0eadf3d](https://app.codecov.io/gh/vturrisi/solo-learn/commit/0eadf3d1587d0e0f6ae6f99bd2b6dc6ebdc2f47f?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) | *This pull request uses carry forward flags. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) to find out more. | [Files](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) | Coverage Δ | | |---|---|---| | [solo/losses/\_\_init\_\_.py](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi#diff-c29sby9sb3NzZXMvX19pbml0X18ucHk=) | `100.00% <100.00%> (ø)` | | | [solo/methods/\_\_init\_\_.py](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi#diff-c29sby9tZXRob2RzL19faW5pdF9fLnB5) | `100.00% <100.00%> (ø)` | | | [solo/methods/ssley.py](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi#diff-c29sby9tZXRob2RzL3NzbGV5LnB5) | `100.00% <100.00%> (ø)` | | | [solo/losses/ssley.py](https://app.codecov.io/gh/vturrisi/solo-learn/pull/378?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi#diff-c29sby9sb3NzZXMvc3NsZXkucHk=) | `75.00% <75.00%> (ø)` | |