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
Categories
Transformer ModelsLicense
MIT LicenseFollow solo-learn
Other Useful Business Software
MicroStation by Bentley Systems is the trusted computer-aided design (CAD) software built specifically for infrastructure design.
MicroStation is the only computer-aided design software for infrastructure design, helping architects and engineers like you bring their vision to life, present their designs to their clients, and deliver their projects to the community.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of solo-learn!