Skip to content

Commit cf02888

Browse files
authored
Fix dygraph dropout seed. test=release/2.0 (#24199)
1 parent efa0007 commit cf02888

File tree

1 file changed

+4
-5
lines changed
  • python/paddle/fluid/layers

1 file changed

+4
-5
lines changed

python/paddle/fluid/layers/nn.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -868,12 +868,11 @@ def get_attrs(prog, dropout_prob, is_test, seed):
868868
if (seed is None or
869869
seed == 0) and default_main_program().random_seed != 0:
870870
seed = default_main_program().random_seed
871-
seed = seed if seed is not None else 0
872871
_is_test = not _dygraph_tracer()._train_mode
873-
out, mask = core.ops.dropout(x, 'dropout_prob', dropout_prob, 'is_test',
874-
_is_test, 'fix_seed', seed is not None,
875-
'seed', seed, 'dropout_implementation',
876-
dropout_implementation)
872+
out, mask = core.ops.dropout(
873+
x, 'dropout_prob', dropout_prob, 'is_test', _is_test, 'fix_seed',
874+
seed is not None, 'seed', seed if seed is not None else 0,
875+
'dropout_implementation', dropout_implementation)
877876
return out
878877

879878
helper = LayerHelper('dropout', **locals())

0 commit comments

Comments
 (0)