From f65e2a7bdb569a14385dcf8ea3d1e20e30cbd16c Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Tue, 5 Dec 2017 16:02:48 +0800
Subject: [PATCH] update epsilon default value in batch_norm

---
 mace/ops/batch_norm.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/ops/batch_norm.h b/mace/ops/batch_norm.h
index 195d9727..3253dcfc 100644
--- a/mace/ops/batch_norm.h
+++ b/mace/ops/batch_norm.h
@@ -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 {
-- 
GitLab