BarlowTwinsConfig

class stable_ssl.joint_embedding.BarlowTwinsConfig(name: str = 'Supervised', backbone_model: str = 'resnet18', sync_batchnorm: bool = False, memory_format: str = 'channels_last', pretrained: bool = False, with_classifier: bool = True, projector: list[int] = <factory>, lambd: str = 0.1)

Bases: JEConfig

Configuration for the BarlowTwins model parameters.

Parameters:

lambd (str) – Lambda parameter for the off-diagonal loss. Default is 0.1.