def update_target_network(self, mm):
        r"""Performs a momentum update of the target network's weights.
        Args:
            mm (float): Momentum used in moving average update.
        """
        for param_q, param_k in zip(
            self.online_encoder.parameters(), self.target_encoder.parameters()
        ):
            param_k.data.mul_(mm).add_(param_q.data, alpha=1.0 - mm)
Performs a momentum update of the target network's weights.
        Args:
            mm (float): Momentum used in moving average update.
        
examples/pytorch/bgrl/model.py