mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

- fixing/aligning CPU implementation of Batch Normalization;

Browse Source
This commit is contained in:
Juan Altmayer Pizzorno 2020-03-23 10:28:48 -04:00
parent 61c9d02ec4
commit c08d808c3d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/blas.c
View File

@ -151,7 +151,7 @@ void normalize_cpu(float *x, float *mean, float *variance, int batch, int filter
for(f = 0; f < filters; ++f){ for(f = 0; f < filters; ++f){
for(i = 0; i < spatial; ++i){ for(i = 0; i < spatial; ++i){
int index = b*filters*spatial + f*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));
} }
} }
} }