byol-torch
is a pytorch implementation of the Boostrap Your Own Latent SSL (self-supervised learning) strategy by Grill, et al. (2020).
- Add support for pretraining on other datasets
- Add KNN (k=20) experimentation regime
- Add data parallel training options for multiple GPU's