def METHOD_NAME(self) -> None:
    """
    Reset all  stats
    """
    self.val: torch.Tensor = None  # type: ignore
    self.sum = torch.tensor(0, dtype=torch.float)
    self.count = torch.tensor(0, dtype=torch.float)
    self.is_distributed = dist.is_available() and dist.is_initialized()