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