VICRegConfig

class stable_ssl.joint_embedding.VICRegConfig(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>, sim_coeff: float = 25, std_coeff: float = 25, cov_coeff: float = 1, epsilon: float = 0.0001)

Bases: JEConfig

Configuration for the VICreg model parameters.