def __init__(self, lmda: float = 5e-3):
        """Initializes the MMCRLoss module with the specified lambda parameter.

        Args:
            lmda: The regularization parameter.

        Raises:
            ValueError: If lmda is less than 0.
        """
        super().__init__()
        if lmda < 0:
            raise ValueError("lmda must be greater than or equal to 0")

        self.lmda = lmda
Initializes the MMCRLoss module with the specified lambda parameter.

        Args:
            lmda: The regularization parameter.

        Raises:
            ValueError: If lmda is less than 0.
        
lightly/loss/mmcr_loss.py