VICReg#
- class stable_ssl.VICReg(config, *args, **kwargs)[source]#
Bases:
JETrainer
VICReg model from [BPL21].
Reference#
[BPL21]Bardes, A., Ponce, J., & LeCun, Y. (2021). VICReg: Variance-Invariance-Covariance Regularization For Self-Supervised Learning. International Conference on Learning Representations (ICLR).