p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!12583 Update description of recompute python api 

From: @ginfung
Reviewed-by: @hwhewei,@zh_qh
Signed-off-by: @zh_qh
This commit is contained in:
mindspore-ci-bot 2021-02-24 17:18:39 +08:00 committed by Gitee
parent 5fadc61222 047e006aab
commit 33d62bce88
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/nn/cell.py
View File

@ -1144,8 +1144,14 @@ class Cell(Cell_):
def recompute(self, mode=True):
"""
Set the cell recomputed. All the primitive in the cell will be set recomputed. If a primitive feeds into a grad
node and is set recomputed, we will compute it again for the grad node after the forward computation.
Set the cell recomputed. All the primitive in the cell will be set recomputed. If a primitive
set recomputed feeds into a gradient node, we will compute it again for the gradient node
after the forward computation.
Note:
If the recompute api of a primtive in this cell is also called, the recompute mode of this
primitive is subject to the recompute api of the primitive.
Args:
mode (bool): Specifies whether the cell is recomputed. Default: True.
"""

5
mindspore/ops/primitive.py
View File

@ -222,8 +222,9 @@ class Primitive(Primitive_):
def recompute(self, mode):
"""
Set the primitive recomputed. If a primitive feeds into a grad node and is set recomputed,
we will compute it again for the grad node after the forward computation.
Set the primitive recomputed. If a primitive set recomputed feeds into a gradient node,
we will compute it again for the gradient node after the forward computation.
Args:
mode (bool): Specifies whether the primitive is recomputed. Default: True.
"""