yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modify comment of register_backward_hook

This commit is contained in:
simson 2020-07-25 16:43:31 +08:00
parent 14ce0afab3
commit 63bb52b408
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

@ -860,7 +860,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`,