Shall we contribute back the lightning implementation

facebookresearch / swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

Other

1.99k stars 280 forks source link

Shall we contribute back the lightning implementation #41

Closed williamFalcon closed 3 years ago

williamFalcon commented 3 years ago

@mathildecaron31 thanks for the help with the lightning implementation.

Would you like us to submit a PR for it?

mathildecaron31 commented 3 years ago

Hi @williamFalcon

Thank you so much for referencing swav in lightning, it was great working with you on that front. I will add the link to the lightning implementation in the README.

I am not quite sure I understand what you mean ? This repo is only for a minimal reproduction of the swav paper.