diff --git a/mace/ops/batch_norm.h b/mace/ops/batch_norm.h index 195d9727e1842db37fc29a5865e6e2785a9838e9..3253dcfc4df41bd7321f058ce061f4cb4938a20b 100644 --- a/mace/ops/batch_norm.h +++ b/mace/ops/batch_norm.h @@ -16,7 +16,7 @@ class BatchNormOp : public Operator { BatchNormOp(const OperatorDef &operator_def, Workspace *ws) : Operator(operator_def, ws), functor_() { functor_.epsilon_ = - OperatorBase::GetSingleArgument("epsilon", static_cast(-1)); + OperatorBase::GetSingleArgument("epsilon", static_cast(1e-4)); } bool Run() override {