Browse Source

Fix the format of note in the recompute api

tags/v1.2.0-rc1
yujianfeng 4 years ago
parent
commit
d1d728ad1c
2 changed files with 5 additions and 3 deletions
  1. +3
    -2
      mindspore/nn/cell.py
  2. +2
    -1
      mindspore/ops/primitive.py

+ 3
- 2
mindspore/nn/cell.py View File

@@ -1151,10 +1151,11 @@ class Cell(Cell_):
intermediate activation computed in forward pass, we will recompute it in backward pass.

Note:

- If the computation involves something like randomization or global variable, the equivalence
is not guaranteed currently.
is not guaranteed currently.
- If the recompute api of a primitive in this cell is also called, the recompute mode of this
primitive is subject to the recompute api of the primitive.
primitive is subject to the recompute api of the primitive.

Args:
mode (bool): Specifies whether the cell is recomputed. Default: True.


+ 2
- 1
mindspore/ops/primitive.py View File

@@ -227,8 +227,9 @@ class Primitive(Primitive_):
pass, we will recompute it in backward pass.

Note:

- If the computation involves something like randomization or global variable, the equivalence
is not guaranteed currently.
is not guaranteed currently.

Args:
mode (bool): Specifies whether the primitive is recomputed. Default: True.


Loading…
Cancel
Save