未验证 提交 5d9df8c8 编写于 作者: L lvmengsi 提交者: GitHub

fix dropout (#18965)

Fix dropout in nn.py
上级 4da1c4f1
...@@ -1558,7 +1558,7 @@ def dropout(x, ...@@ -1558,7 +1558,7 @@ def dropout(x,
'dropout_prob': dropout_prob, 'dropout_prob': dropout_prob,
'is_test': is_test, 'is_test': is_test,
'fix_seed': seed is not None, 'fix_seed': seed is not None,
'seed': seed, 'seed': seed if seed is not None else 0,
'dropout_implementation': dropout_implementation, 'dropout_implementation': dropout_implementation,
}) })
return out return out
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册