未验证 提交 f44add7b 编写于 作者: W weishengying 提交者: GitHub

modify the fix_seed attribute in dropout op is a def attribute.test=develop (#38100)

上级 8198cad7
...@@ -88,8 +88,8 @@ class DropoutOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -88,8 +88,8 @@ class DropoutOpMaker : public framework::OpProtoAndCheckerMaker {
"training. Setting this flag to true is only useful in " "training. Setting this flag to true is only useful in "
"unittest or for debug that always the same output units " "unittest or for debug that always the same output units "
"will be dropped.") "will be dropped.")
.SetDefault(false) .SetDefault(false);
.AsExtra();
AddAttr<int>("seed", "Dropout random seed.").SetDefault(0).AsExtra(); AddAttr<int>("seed", "Dropout random seed.").SetDefault(0).AsExtra();
AddAttr<std::string>( AddAttr<std::string>(
"dropout_implementation", "dropout_implementation",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册