diff --git a/mindspore/nn/cell.py b/mindspore/nn/cell.py index f2f72bc187a..621ee06d089 100755 --- a/mindspore/nn/cell.py +++ b/mindspore/nn/cell.py @@ -1162,10 +1162,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. diff --git a/mindspore/ops/primitive.py b/mindspore/ops/primitive.py index 8fe67dc10b0..c88bda5073b 100644 --- a/mindspore/ops/primitive.py +++ b/mindspore/ops/primitive.py @@ -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.