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
$300 in Free Credit Towards Top Cloud Services Icon
$300 in Free Credit Towards Top Cloud Services

Build VMs, containers, AI, databases, storage—all in one place.

Start your project in minutes. After credits run out, 20+ products include free monthly usage. Only pay when you're ready to scale.
Get Started
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