提交 15ae3702 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5063 modify sgd and momentum and WithGradCell comments

Merge pull request !5063 from lijiaqi/momentum_and_sgd
......@@ -82,7 +82,7 @@ class WithGradCell(Cell):
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
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:
Run in PyNative mode.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册