← all repositories

lucidrains/byol-pytorch

A PyTorch implementation of BYOL, a self-supervised learning technique for training neural networks on unlabeled image data.

1.9k stars Python ML Frameworks
byol-pytorch
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

Implements the BYOL self-supervised learning algorithm from DeepMind in PyTorch. The library provides a wrapper module that can be applied to any image-based neural network to enable training on unlabeled data through prediction of latent representations. It eliminates the need for labeled data or contrastive learning approaches, achieving competitive results with SimCLR using a simpler methodology.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.