SSL Methods

SSL Methods#

This module contains the following SSL methods:

SimCLR(config, *args, **kwargs)

SimCLR model from [CKNH20].

BarlowTwins(config, *args, **kwargs)

BarlowTwins model from [ZJM+21].

VICReg(config, *args, **kwargs)

VICReg model from [BPL21].

WMSE(config, *args, **kwargs)

Whitening Mean Squared Error (WMSE) model from [ESSS21].