typo

Lightning-AI · Mar 18, 2022 · 5f95ceb · 5f95ceb
1 parent d3e8d37
commit 5f95ceb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmetrics/classification/kl_divergence.py b/torchmetrics/classification/kl_divergence.py
@@ -80,7 +80,7 @@ class KLDivergence(Metric):
  def __init__(
  self,
  log_prob: bool = False,
- reduction: Literal["elementwise_mean", "sum", "none", None] = "mean",
+ reduction: Literal["mean", "sum", "none", None] = "mean",
  compute_on_step: Optional[bool] = None,
  **kwargs: Dict[str, Any],
  ) -> None: