Batchnorm eps fix, works on jetpack 4.3

2020-01-15 18:06:02 +01:00
parent 57d7743f7e
commit c32a0be257
5 changed files with 13 additions and 9 deletions
@@ -24,6 +24,8 @@ enum layerType_t {
    LAYER_YOLO
 };

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