Python mxnet.ndarray.adam_update用法及代碼示例

用法:
mxnet.ndarray.adam_update(weight=None, grad=None, mean=None, var=None, lr=_Null, beta1=_Null, beta2=_Null, epsilon=_Null, wd=_Null, rescale_grad=_Null, clip_gradient=_Null, lazy_update=_Null, out=None, name=None, **kwargs)

參數：

weight：(NDArray) - 重量
grad：(NDArray) - 坡度
mean：(NDArray) - 移動均值
var：(NDArray) - 移動方差
lr：(float, required) - 學習率
beta1：(float, optional, default=0.899999976) - 一階矩估計的衰減率。
beta2：(float, optional, default=0.999000013) - 二階矩估計的衰減率。
epsilon：(float, optional, default=9.99999994e-09) - 數值穩定性的小常數。
wd：(float, optional, default=0) - 權重衰減通過懲罰大權重的正則化項來增強目標函數。懲罰與每個權重大小的平方成比例。
rescale_grad：(float, optional, default=1) - 將漸變重新縮放為 grad = rescale_grad*grad。
clip_gradient：(float, optional, default=-1) - 將漸變剪裁到 [-clip_gradient, clip_gradient] 的範圍內如果clip_gradient <= 0，漸變剪裁被關閉。畢業 = 最大(最小(畢業，clip_gradient)，-clip_gradient)。
lazy_update：(boolean, optional, default=1) - 如果為 true，如果梯度的 stype 為 row_sparse 並且所有 w、m 和 v 具有相同的 stype，則應用延遲更新
out：(NDArray, optional) - 輸出 NDArray 來保存結果。

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

Adam 優化器的更新函數。 Adam 被視為 AdaGrad 的推廣。

Adam 更新包括以下步驟，其中 g 表示梯度，m、v 是一階和二階矩估計(均值和方差)。

\[\begin{split}g_t = \nabla J(W_{t-1})\\ m_t = \beta_1 m_{t-1} + (1 - \beta_1) g_t\\ v_t = \beta_2 v_{t-1} + (1 - \beta_2) g_t^2\\ W_t = W_{t-1} - \alpha \frac{ m_t }{ \sqrt{ v_t } + \epsilon }\end{split}\]

它使用以下方法更新權重：

m = beta1*m + (1-beta1)*grad
v = beta2*v + (1-beta2)*(grad**2)
w += - learning_rate * m / (sqrt(v) + epsilon)

但是，如果grad的存儲類型為row_sparse，lazy_update為True，並且權重的存儲類型與m和v相同，則隻更新索引出現在grad.indices中的row slice(對於w，m五)：

for row in grad.indices:
    m[row] = beta1*m[row] + (1-beta1)*grad[row]
    v[row] = beta2*v[row] + (1-beta2)*(grad[row]**2)
    w[row] += - learning_rate * m[row] / (sqrt(v[row]) + epsilon)

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.adam_update。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：