提交 ede33c62 编写于 作者: C ceci3

fix formula in dropout

上级 8468c393
......@@ -1316,7 +1316,7 @@ def dropout(x,
1. downgrade_in_infer(default), downgrade the outcome at inference
- train: out = input * mask
- inference: out = input * dropout_prob
- inference: out = input * (1 - dropout_prob)
(mask is a tensor same shape with input, value is 0 or 1
ratio of 0 is dropout_prob)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册