update the formula of smooth_l1_loss (#1986)

2025-10-31 07:58:14 +08:00 · 2025-03-21 21:25:23 +08:00
parent 7b7e2352cd
commit 65a38c452b
1 changed files with 1 additions and 1 deletions
--- a/python/mlx/nn/losses.py
+++ b/python/mlx/nn/losses.py
@@ -352,7 +352,7 @@ def smooth_l1_loss(
    .. math::

      l = \begin{cases}
-            0.5 (x - y)^2, & \text{if } (x - y) < \beta \\
+            0.5 (x - y)^2 / \beta, & \text{if } |x - y| < \beta \\
            |x - y| - 0.5 \beta, & \text{otherwise}
          \end{cases}