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

6
src/connected_layer.c
View File

@ -148,6 +148,12 @@ void forward_connected_layer_gpu(connected_layer l, network_state state)
float * c = l.output_gpu;
gemm_ongpu(0,0,m,n,k,1,a,k,b,n,1,c,n);
activate_array_ongpu(l.output_gpu, l.outputs*l.batch, l.activation);
/*
cuda_pull_array(l.output_gpu, l.output, l.outputs*l.batch);
float avg = mean_array(l.output, l.outputs*l.batch);
printf("%f\n", avg);
*/
}
void backward_connected_layer_gpu(connected_layer l, network_state state)