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

!34672 Fix format for set_dump api introduce 

Merge pull request !34672 from maning202007/code_docs_dump_api
This commit is contained in:
i-robot 2022-05-20 02:02:30 +00:00 committed by Gitee
parent 9b0d37c7e7 3bc41e4d83
commit 98badab6a7
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/api/api_python/mindspore/mindspore.set_dump.rst
View File

@ -14,7 +14,7 @@ mindspore.set_dump
- 此API只在Ascend后端的图模式有效。
- 此API只支持训练开始前调用。如果在训练过程中调用这个API，可能不会有效果。
- 使用 `set_dump(Cell, True)` 后，Cell正向计算和反向计算（梯度运算产生的计算）中的算子会被Dump。
- 对于 :class:`mindspore.nn.SoftMaxCrossEntropyWithLogits` 层，正向计算和反向计算使用同一组算子。因此，只能看到反向计算中的Dump数据。请注意，当使用 `sparse=True` 和 `reduce=“mean”` 初始化时， :class:`mindspore.nn.SoftmaxCrossEntropyWithLogits` 层也将在内部使用这些算子。
- 对于 :class:`mindspore.nn.SoftmaxCrossEntropyWithLogits` 层，正向计算和反向计算使用同一组算子。因此，只能看到反向计算中的Dump数据。请注意，当使用 `sparse=True` 和 `reduce=“mean”` 初始化时， :class:`mindspore.nn.SoftmaxCrossEntropyWithLogits` 层也将在内部使用这些算子。
**参数：**