Browse Source

modify annotations for inputs of sparse adam

tags/v0.5.0-beta
wangnan39@huawei.com 5 years ago
parent
commit
9fe3856b9f
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ops/operations/nn_ops.py

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

@@ -2777,7 +2777,7 @@ class SparseApplyAdam(PrimitiveWithInfer):
- **m** (Parameter) - The 1st moment vector in the updating formula. Has the same type as `var`. With
float32 data type.
- **v** (Parameter) - The 2nd moment vector in the updating formula. Mean square gradients,
has the same type as `var`. With float32 data type.
has the same type as `var`. With float32 data type.
- **beta1_power** (Tensor) - :math:`beta_1^t` in the updating formula. With float32 data type.
- **beta2_power** (Tensor) - :math:`beta_2^t` in the updating formula. With float32 data type.
- **lr** (Tensor) - :math:`l` in the updating formula. With float32 data type.
@@ -2911,7 +2911,7 @@ class SparseApplyLazyAdam(PrimitiveWithInfer):
- **m** (Parameter) - The 1st moment vector in the updating formula. Has the same type as `var`. With
float32 data type.
- **v** (Parameter) - The 2nd moment vector in the updating formula. Mean square gradients,
has the same type as `var`. With float32 data type.
has the same type as `var`. With float32 data type.
- **beta1_power** (Tensor) - :math:`beta_1^t` in the updating formula. With float32 data type.
- **beta2_power** (Tensor) - :math:`beta_2^t` in the updating formula. With float32 data type.
- **lr** (Tensor) - :math:`l` in the updating formula. With float32 data type.


Loading…
Cancel
Save