Fix dygraph dropout seed. test=develop (#24177)

revert-24314-dev/fix_err_msg
Guo Sheng 5 years ago committed by GitHub
parent 9b851ba216
commit 663eca451d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -983,12 +983,11 @@ def dropout(x,
if (seed is None or
seed == 0) and default_main_program().random_seed != 0:
seed = default_main_program().random_seed
seed = seed if seed is not None else 0
_is_test = not _dygraph_tracer()._train_mode
out, mask = core.ops.dropout(x, 'dropout_prob', dropout_prob, 'is_test',
_is_test, 'fix_seed', seed is not None,
'seed', seed, 'dropout_implementation',
dropout_implementation)
out, mask = core.ops.dropout(
x, 'dropout_prob', dropout_prob, 'is_test', _is_test, 'fix_seed',
seed is not None, 'seed', seed if seed is not None else 0,
'dropout_implementation', dropout_implementation)
return out
helper = LayerHelper('dropout', **locals())

Loading…
Cancel
Save