def _initialize_weights(self) -> None:
        """Initializes weights for the backbone components."""

        # Initialize the patch embedding layer like a linear layer instead of conv
        # layer.
        w = self.conv_proj.weight.data
        torch.nn.init.xavier_uniform_(w.view([w.shape[0], -1]))

        # Initialize the class token.
        torch.nn.init.normal_(self.class_token, std=0.02)

        self.encoder._initialize_weights()
        _initialize_linear_layers(self)
Initializes weights for the backbone components.
lightly/models/modules/masked_autoencoder.py