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

        torch.nn.init.normal_(self.mask_token, std=0.02)
        utils.initialize_2d_sine_cosine_positional_embedding(
            pos_embedding=self.decoder_pos_embed, has_class_token=True
        )
        self.apply(init_weights)
Initializes weights for the decoder components.
lightly/models/modules/masked_autoencoder_timm.py