search

lucidrains / byol-pytorch

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

Add use_momentum support for Lightning #45

Closed mark-selyaeff closed 3 years ago

lucidrains commented 3 years ago

@mark-selyaeff awesome!