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

Faster batch normalization

Browse Source
This commit is contained in:
Joseph Redmon
2015-12-07 17:18:04 -08:00
parent b4d81aa7bd
commit c2738835f0
7 changed files with 149 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/convolutional_layer.c
View File

@@ -86,9 +86,8 @@ convolutional_layer make_convolutional_layer(int batch, int h, int w, int c, int
}
l.mean = calloc(n, sizeof(float));
l.spatial_mean = calloc(n*l.batch, sizeof(float));
l.variance = calloc(n, sizeof(float));
l.rolling_mean = calloc(n, sizeof(float));
l.rolling_variance = calloc(n, sizeof(float));
}
@@ -114,12 +113,6 @@ convolutional_layer make_convolutional_layer(int batch, int h, int w, int c, int
l.rolling_mean_gpu = cuda_make_array(l.mean, n);
l.rolling_variance_gpu = cuda_make_array(l.variance, n);
l.spatial_mean_gpu = cuda_make_array(l.spatial_mean, n*l.batch);
l.spatial_variance_gpu = cuda_make_array(l.spatial_mean, n*l.batch);
l.spatial_mean_delta_gpu = cuda_make_array(l.spatial_mean, n*l.batch);
l.spatial_variance_delta_gpu = cuda_make_array(l.spatial_mean, n*l.batch);
l.mean_delta_gpu = cuda_make_array(l.mean, n);
l.variance_delta_gpu = cuda_make_array(l.variance, n);