Batchnorm eps fix, works on jetpack 4.3

2020-01-15 18:06:02 +01:00
parent 7233b065a8
commit 33844c1ab2
5 changed files with 21 additions and 17 deletions
@@ -27,6 +27,8 @@ enum layerType_t
    LAYER_YOLO
 };

+#define TKDNN_BN_MIN_EPSILON 1e-5
+
 /** 
    Simple layer Father class
 */