def __init__(
        self,
        tcr_eps: float = 0.2,
        inv_coef: float = 200.0,
    ) -> None:
        """Initializes the EMPSSLoss module.

        Args:
            tcr_eps:
                Total coding rate (TCR) epsilon.
            inv_coff:
                Coefficient for the invariance loss.
        """
        super().__init__()
        self.tcr_eps = tcr_eps
        self.inv_coef = inv_coef
Initializes the EMPSSLoss module.

        Args:
            tcr_eps:
                Total coding rate (TCR) epsilon.
            inv_coff:
                Coefficient for the invariance loss.
        
lightly/loss/emp_ssl_loss.py