提交 4eadfc07 编写于 作者: X Xinghai Sun

Correct typos for dropout operator.

上级 963a4f3c
...@@ -41,9 +41,7 @@ class DropoutOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -41,9 +41,7 @@ class DropoutOpMaker : public framework::OpProtoAndCheckerMaker {
DropoutOpMaker(framework::OpProto *proto, DropoutOpMaker(framework::OpProto *proto,
framework::OpAttrChecker *op_checker) framework::OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) { : OpProtoAndCheckerMaker(proto, op_checker) {
AddAttr<float>("dropout_prob", AddAttr<float>("dropout_prob", "Probability for dropping out units.")
"Probability of randomly setting elements "
"to zero.")
.SetDefault(.5f); .SetDefault(.5f);
AddAttr<int>("seed", "Dropout random seed.").SetDefault(0); AddAttr<int>("seed", "Dropout random seed.").SetDefault(0);
AddInput("X", "The input of dropout op."); AddInput("X", "The input of dropout op.");
...@@ -56,7 +54,7 @@ Dropout Operator. ...@@ -56,7 +54,7 @@ Dropout Operator.
"Dropout" refers to randomly dropping out units in a nerual network. It is a "Dropout" refers to randomly dropping out units in a nerual network. It is a
regularization technique for reducing overfitting by preventing neuron regularization technique for reducing overfitting by preventing neuron
co-adaption during training. The dropout operator randomly set (according to co-adaption during training. The dropout operator randomly set (according to
the given dropout probability) the output of some units to zero, while others the given dropout probability) the outputs of some units to zero, while others
being set to their inputs. being set to their inputs.
)DOC"); )DOC");
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册