A library of self-supervised methods for visual representation learning powered by Pytorch Lightning. A library of self-supervised methods for unsupervised visual representation learning powered by PyTorch Lightning. We aim at providing SOTA self-supervised methods in a comparable environment while, at the same time, implementing training tricks. The library is self-contained, but it is possible to use the models outside of solo-learn.

Features

  • Increased data processing speed by up to 100% using Nvidia Dali
  • Flexible augmentations
  • Online linear evaluation via stop-gradient for easier debugging and prototyping (optionally available for the momentum backbone as well)
  • Standard offline linear evaluation
  • Online and offline K-NN evaluation
  • Automatic feature space visualization with UMAP

Project Samples

solo-learn Screenshot 1

Project Activity

See All Activity >

Categories

Transformer Models

License

MIT License

Follow solo-learn

solo-learn Web Site

Other Useful Business Software
Gemini 3 and 200+ AI Models on One Platform Icon
Gemini 3 and 200+ AI Models on One Platform

Access Google's best plus Claude, Llama, and Gemma. Fine-tune and deploy from one console.

Build, govern, and optimize agents and models with Gemini Enterprise Agent Platform.
Start Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of solo-learn!

Additional Project Details

Programming Language

Python

Related Categories

Python Transformer Models

Registered

2023-04-21
Report inappropriate content