From 2e471f85053fbf4c7f48093d51ffdfbacc4b4cff Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Fri, 29 May 2020 17:14:19 +0800 Subject: [PATCH] fixed doc for ApplyRMSProp --- mindspore/ops/operations/nn_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index b36c65cf82..2c6121bc4d 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -1705,8 +1705,8 @@ class ApplyRMSProp(PrimitiveWithInfer): - **var** (Tensor) - Weights to be update. - **mean_square** (Tensor) - Mean square gradients, must have the same type as `var`. - **moment** (Tensor) - Delta of `var`, must have the same type as `var`. - - **grad** (Tensor) - Gradients, must have the same type as `var`. - **learning_rate** (Union[Number, Tensor]) - Learning rate. + - **grad** (Tensor) - Gradients, must have the same type as `var`. - **decay** (float) - Decay rate. - **momentum** (float) - Momentum. - **epsilon** (float) - Ridge term.