def __init__(self, offdiag_lambda=1.0, eps=1e-12):
        """
        Args:
            offdiag_lambda: trade-off parameter
            eps: shift for the varience (var + eps)
        """
        super().__init__()
        self.offdiag_lambda = offdiag_lambda
        self.eps = eps

        Args:
            offdiag_lambda: trade-off parameter
            eps: shift for the varience (var + eps)
        
catalyst/contrib/losses/contrastive.py