solo-learn
solo-learn copied to clipboard
Implements SSL-EY
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.
hey @jameschapman19 thanks for contributing. I'll try to start checking your code soon.
@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.
@jameschapman19 no worries, take your time :) and ping me when you are done
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 is93.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 | Coverage Δ | *Carryforward flag | |
---|---|---|---|
cpu | 80.02% <93.75%> (+0.15%) |
:arrow_up: | |
dali | 38.81% <ø> (ø) |
Carriedforward from 0eadf3d |
*This pull request uses carry forward flags. Click here to find out more.
Files | Coverage Δ | |
---|---|---|
solo/losses/__init__.py | 100.00% <100.00%> (ø) |
|
solo/methods/__init__.py | 100.00% <100.00%> (ø) |
|
solo/methods/ssley.py | 100.00% <100.00%> (ø) |
|
solo/losses/ssley.py | 75.00% <75.00%> (ø) |