WMSE

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

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.

initialize_modules()

Initialize the modules required for the model.