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

!3487 Modify comment of register_backward_hook 

Merge pull request !3487 from Simson/push-to-opensource
This commit is contained in:
mindspore-ci-bot 2020-07-25 19:11:36 +08:00 committed by Gitee
parent 7478c26b75 61d961ddec
commit 57c00a77f1
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/cell.py
View File

@ -862,7 +862,7 @@ class Cell:
def register_backward_hook(self, fn):
"""
Set the cell backward hook function.
Set the cell backward hook function. Note that this function is only supported in Pynative Mode.
Note:
fn should be defined as following code shows, `cell_name` is the name of registered cell,

3
mindspore/ops/operations/debug_ops.py
View File

@ -251,7 +251,8 @@ class InsertGradientOf(PrimitiveWithInfer):
class HookBackward(PrimitiveWithInfer):
"""
Used as tag to hook gradient in intermediate variables.
Used as tag to hook gradient in intermediate variables. Note that this function
is only supported in Pynative Mode.
Note:
The hook function should be defined like `hook_fn(grad) -> Tensor or None`,