solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
-
Updated
Nov 4, 2024 - Python
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
Framework for training and evaluating self-supervised learning methods for speaker verification.
This a tensorflow implementation of VICReg - a self-supervised learning architecture that prevents collapse in an intuitive manner using a loss function that 1. maintains the variance of each embedding over a batch above a threshold and 2. decorrelating pairs of embeddings over a batch and attracting them to 0. Training was done using TPU on colab
A VIcReg Implementation in pytorch
Add a description, image, and links to the vicreg topic page so that developers can more easily learn about it.
To associate your repository with the vicreg topic, visit your repo's landing page and select "manage topics."