提交 f65e2a7b 编写于 作者: Y yejianwu

update epsilon default value in batch_norm

上级 d4969fc0
......@@ -16,7 +16,7 @@ class BatchNormOp : public Operator<D, T> {
BatchNormOp(const OperatorDef &operator_def, Workspace *ws)
: Operator<D, T>(operator_def, ws), functor_() {
functor_.epsilon_ =
OperatorBase::GetSingleArgument<float>("epsilon", static_cast<float>(-1));
OperatorBase::GetSingleArgument<float>("epsilon", static_cast<float>(1e-4));
}
bool Run() override {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册