Browse Source

!2158 modify annotations for inputs of sparse adam

Merge pull request !2158 from wangnan39/modify_annotations_of_sparse_adam
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
309406067c
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