WMSE#
- class stable_ssl.WMSE(config, *args, **kwargs)[source]#
Bases:
JETrainer
Whitening Mean Squared Error (WMSE) model from [ESSS21].
Reference#
[ESSS21]Ermolov, A., Siarohin, A., Sangineto, E., & Sebe, N. (2021). Whitening for self-supervised representation learning. In International conference on machine learning (pp. 3015-3024). PMLR.