提交 1ec7e0d9 编写于 作者: 叶剑武

Merge branch 'update_epsilon_in_batch_norm' into 'master'

update epsilon default value in batch_norm

See merge request !148
......@@ -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.
先完成此消息的编辑!
想要评论请 注册