WMSEConfig

class stable_ssl.joint_embedding.WMSEConfig(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>, w_iter: float = 1, w_eps: float = 0)

Bases: JEConfig

Configuration for the WMSE model parameters.