mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!3493 Modify comment of register_backward_hook [r0.6] 

Merge pull request !3493 from Simson/push-to-r06
This commit is contained in:
mindspore-ci-bot 2020-07-27 09:30:26 +08:00 committed by Gitee
parent 7f6f140d94 63bb52b408
commit fdf198eee9
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`,