diff --git a/docs/api/api_python/nn/mindspore.nn.Lamb.rst b/docs/api/api_python/nn/mindspore.nn.Lamb.rst
index d4b9b33f3b0..280617fe747 100644
--- a/docs/api/api_python/nn/mindspore.nn.Lamb.rst
+++ b/docs/api/api_python/nn/mindspore.nn.Lamb.rst
@@ -37,7 +37,7 @@ mindspore.nn.Lamb
              /(\sqrt{\hat{\boldsymbol{v}}_{t}}+\epsilon) \\
             &\hspace{5mm}\boldsymbol{w}_{t+1}^{(i)}=\boldsymbol{w}_{t}^{(i)}- \gamma_{t}
              \frac{\boldsymbol{\phi}\left(\left\|\boldsymbol{w}_{t}^{(i)}\right\|\right)}
-             {\left\|\boldsymbol{w}_{t}^{(i)}+\lambda \boldsymbol{w}_{t}^{(i)}\right\|}\left(\boldsymbol{r}_{t}^{(i)}+
+             {\left\|\boldsymbol{r}_{t}^{(i)}+\lambda \boldsymbol{w}_{t}^{(i)}\right\|}\left(\boldsymbol{r}_{t}^{(i)}+
              \lambda \boldsymbol{w}_{t}^{(i)}\right) \\
             &\textbf{end for} \\[-1.ex]
             &\newline
diff --git a/mindspore/python/mindspore/nn/optim/lamb.py b/mindspore/python/mindspore/nn/optim/lamb.py
index fc7730c527a..0a5480d75de 100755
--- a/mindspore/python/mindspore/nn/optim/lamb.py
+++ b/mindspore/python/mindspore/nn/optim/lamb.py
@@ -113,7 +113,7 @@ class Lamb(Optimizer):
              /(\sqrt{\hat{\boldsymbol{v}}_{t}}+\epsilon) \\
             &\hspace{5mm}\boldsymbol{w}_{t+1}^{(i)}=\boldsymbol{w}_{t}^{(i)}- \gamma_{t}
              \frac{\boldsymbol{\phi}\left(\left\|\boldsymbol{w}_{t}^{(i)}\right\|\right)}
-             {\left\|\boldsymbol{w}_{t}^{(i)}+\lambda \boldsymbol{w}_{t}^{(i)}\right\|}\left(\boldsymbol{r}_{t}^{(i)}+
+             {\left\|\boldsymbol{r}_{t}^{(i)}+\lambda \boldsymbol{w}_{t}^{(i)}\right\|}\left(\boldsymbol{r}_{t}^{(i)}+
              \lambda \boldsymbol{w}_{t}^{(i)}\right) \\
             &\textbf{end for} \\[-1.ex]
             &\newline