def METHOD_NAME(self, current: torch.Tensor) -> Tuple[bool, str]:
    should_stop, reason = super().METHOD_NAME(current)
    if should_stop:
        self.early_stopping_reason = reason
    return should_stop, reason