From 66ee3bb3bcea3a409f1764a61d9f15a5f7669546 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E7=BF=BC=E7=BF=94?= <836250617@qq.com>
Date: Mon, 2 Sep 2024 04:31:07 +0800
Subject: [PATCH] Update huber.rs (#2232)

---
 crates/burn-core/src/nn/loss/huber.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/burn-core/src/nn/loss/huber.rs b/crates/burn-core/src/nn/loss/huber.rs
index 8b227b0a4..55d57d65c 100644
--- a/crates/burn-core/src/nn/loss/huber.rs
+++ b/crates/burn-core/src/nn/loss/huber.rs
@@ -37,7 +37,7 @@ impl HuberLossConfig {
 /// The loss for each element of the residuals `r = targets - predictions` is given by
 ///
 /// ```text
-/// L(r) = 0.5 * x^2                  if |r| <= d
+/// L(r) = 0.5 * r^2                  if |r| <= d
 /// L(r) = 0.5 * d^2 + d * (|r| - d)  if |r| >  d
 /// ```
 ///