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
Stop Cyber Threats with VM-Series Next-Gen Firewall on Azure Icon
Stop Cyber Threats with VM-Series Next-Gen Firewall on Azure

Native application identity and user-based security for your Azure cloud

Gain integrated visibility across all traffic in a single pass. Deploy Palo Alto Networks VM-Series to determine application identity and content while automating security policy updates via rich APIs.
Get a free trial
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