VICReg

class stable_ssl.VICReg(config, *args, **kwargs)

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).