def _momentum_update(self, m: float = 0.999) -> None:
        """Performs the momentum update for the backbone and projection head."""
        _do_momentum_update(
            self.momentum_backbone.parameters(),
            self.backbone.parameters(),
            m=m,
        )
        _do_momentum_update(
            self.momentum_projection_head.parameters(),
            self.projection_head.parameters(),
            m=m,
        )
Performs the momentum update for the backbone and projection head.
lightly/models/_momentum.py