Closed shivam-aggarwal closed 3 years ago
@shivam-aggarwal Hi Shivam! I've got it fixed in 0.5.6 https://github.com/lucidrains/byol-pytorch/commit/2aa84ee18fafecaf35637da4657f92619e83876d
@lucidrains Thank you so much! Thanks for providing a neat implementation!
As mentioned in the readme
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 .