def __init__(
        self,
        temperature: float = 0.1,
        sigma: float = 0.5,
        gather_distributed: bool = False,
    ):
        """Initializes the DCLWLoss module.

        Args:
            temperature:
                Similarities are scaled by inverse temperature.
            sigma:
                Applies inverse scaling in the weighting function.
            gather_distributed:
                If True, negatives from all GPUs are gathered before the
                loss calculation.
        """
        super().__init__(
            temperature=temperature,
            weight_fn=partial(negative_mises_fisher_weights, sigma=sigma),
            gather_distributed=gather_distributed,
        )
Initializes the DCLWLoss module.

        Args:
            temperature:
                Similarities are scaled by inverse temperature.
            sigma:
                Applies inverse scaling in the weighting function.
            gather_distributed:
                If True, negatives from all GPUs are gathered before the
                loss calculation.
        
lightly/loss/dcl_loss.py