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

!5063 modify sgd and momentum and WithGradCell comments 

Merge pull request !5063 from lijiaqi/momentum_and_sgd
This commit is contained in:
mindspore-ci-bot 2020-08-24 20:23:23 +08:00 committed by Gitee
parent 776d003412 d957309957
commit 15ae3702f9
3 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/wrap/cell_wrapper.py
View File

@ -82,7 +82,7 @@ class WithGradCell(Cell):
Wraps the network with backward cell to compute gradients. A network with a loss function is necessary Wraps the network with backward cell to compute gradients. A network with a loss function is necessary
as argument. If loss function in None, the network must be a wrapper of network and loss function. This as argument. If loss function in None, the network must be a wrapper of network and loss function. This
Cell accepts *inputs as inputs and returns gradients for each trainable parameter. Cell accepts '*inputs' as inputs and returns gradients for each trainable parameter.
Note: Note:
Run in PyNative mode. Run in PyNative mode.