lucidrains / byol-pytorch

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
MIT License
1.76k stars 248 forks source link

0.5.5 version no longer returns projections. #57

Closed shivam-aggarwal closed 3 years ago

shivam-aggarwal commented 3 years ago

As mentioned in the readme

projection, embedding = learner(imgs, return_embedding = True)

Users should be able to get projections and embedding, with 0.5.5 users can no longer do so. Note : It's happening because of this .

lucidrains commented 3 years ago

@shivam-aggarwal Hi Shivam! I've got it fixed in 0.5.6 https://github.com/lucidrains/byol-pytorch/commit/2aa84ee18fafecaf35637da4657f92619e83876d

shivam-aggarwal commented 3 years ago

@lucidrains Thank you so much! Thanks for providing a neat implementation!