← all repositories

vturrisi/solo-learn

A PyTorch Lightning library providing state-of-the-art self-supervised learning methods for visual representation learning.

1.6k stars Python ML FrameworksComputer Vision
solo-learn
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

solo-learn is a self-contained library implementing various self-supervised learning algorithms (BYOL, SimCLR, MoCo, DINO, MAE, Barlow Twins, VICReg, SimSiam, SwAV) for unsupervised visual representation learning. It uses PyTorch Lightning for training, supports NVIDIA DALI for data loading, and allows using the trained backbones outside the library. The project provides SOTA implementations with training tricks and supports finetuning, mixing augmentations, and checkpointing.

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