From f329b89acb277fba5693e2a0415513ee7ae78c70 Mon Sep 17 00:00:00 2001 From: thnkinbtfly <70014488+thnkinbtfly@users.noreply.github.com> Date: Fri, 11 Dec 2020 23:26:45 +0900 Subject: [PATCH] Use correct BN MIN Epsilon --- include/tkDNN/Layer.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/tkDNN/Layer.h b/include/tkDNN/Layer.h index 25c4565..9330f8d 100644 --- a/include/tkDNN/Layer.h +++ b/include/tkDNN/Layer.h @@ -32,7 +32,7 @@ enum layerType_t { LAYER_YOLO }; -#define TKDNN_BN_MIN_EPSILON 1e-5 +#define TKDNN_BN_MIN_EPSILON 1e-6 /** Simple layer Father class