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
Gen AI apps are built with MongoDB Atlas Icon
Gen AI apps are built with MongoDB Atlas

Build gen AI apps with an all-in-one modern database: MongoDB Atlas

MongoDB Atlas provides built-in vector search and a flexible document model so developers can build, scale, and run gen AI apps without stitching together multiple databases. From LLM integration to semantic search, Atlas simplifies your AI architecture—and it’s free to get started.
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