提交 3305c876 编写于 作者: Y yangyaming

Add more comment.

上级 fb56a181
...@@ -71,15 +71,18 @@ X to Y. Different from MSE loss, Huber loss is more robust for outliers. The ...@@ -71,15 +71,18 @@ X to Y. Different from MSE loss, Huber loss is more robust for outliers. The
shape of X and Y are [batch_size, 1]. The equation is: shape of X and Y are [batch_size, 1]. The equation is:
$$ $$
Out_{\delta}(i, x, y) = Out_{\delta}(X, Y)_i =
\begin{cases} \begin{cases}
0.5 * (Input(i, y) - Input(i, x))^2, 0.5 * (Y_i - X_i)^2,
\quad |Input(i, y) - Input(i, x)| \leq \delta \\ \quad |Y_i - X_i| \leq \delta \\
\delta * (|Input(i, y) - Input(i, x)| - 0.5 * \delta), \delta * (|Y_i - X_i| - 0.5 * \delta),
\quad otherwise \quad otherwise
\end{cases} \end{cases}
$$ $$
In the above equation, $Out_\delta(X, Y)_i$, $X_i$ and $Y_i$ represent the ith
element of Out, X and Y.
)DOC"); )DOC");
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册