- adjusted BatchNorm epsilon of normalize_cpu to be the same as

normalize_gpu's kernel;
2020-04-30 14:04:54 -04:00 · 2020-04-30 14:04:54 -04:00 · a34682e67c
parent 6cbb75d10b
commit a34682e67c
1 changed files with 1 additions and 1 deletions
--- a/src/blas.c
+++ b/src/blas.c
@ -289,7 +289,7 @@ void normalize_cpu(float *x, float *mean, float *variance, int batch, int filter
        for(f = 0; f < filters; ++f){
            for(i = 0; i < spatial; ++i){
                int index = b*filters*spatial + f*spatial + i;
-                x[index] = (x[index] - mean[f])/(sqrt(variance[f] + .000001f));
+                x[index] = (x[index] - mean[f])/(sqrt(variance[f] + .00001f));
            }
        }
    }