Browse Source

fixed doc for ApplyRMSProp

tags/v0.5.0-beta
jiangjinsheng 6 years ago
parent
commit
2e471f8505
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/operations/nn_ops.py

+ 1
- 1
mindspore/ops/operations/nn_ops.py View File

@@ -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.


Loading…
Cancel
Save