Skip to content

dvdhs/byol_torch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

✨ byol-torch

byol-torch is a pytorch implementation of the Boostrap Your Own Latent SSL (self-supervised learning) strategy by Grill, et al. (2020).

TODO

  • Add support for pretraining on other datasets
  • Add KNN (k=20) experimentation regime
  • Add data parallel training options for multiple GPU's

About

A pytorch implementation of BYOL

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages