From 9fe3856b9f8244991464b8f7c0034266fb616243 Mon Sep 17 00:00:00 2001 From: "wangnan39@huawei.com" Date: Tue, 16 Jun 2020 11:29:34 +0800 Subject: [PATCH] modify annotations for inputs of sparse adam --- mindspore/ops/operations/nn_ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 49e318169e..00ebc17f5f 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -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.