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

Changes to make routing work better

Browse Source
This commit is contained in:
Joseph Redmon
2015-07-21 16:09:33 -07:00
parent e56d1eff13
commit d00f0a1ccd
18 changed files with 29 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/connected_layer.c
View File

@ -103,7 +103,7 @@ void backward_connected_layer(connected_layer l, network_state state)
b = l.weights;
c = state.delta;
if(c) gemm(0,1,m,n,k,1,a,k,b,k,0,c,n);
if(c) gemm(0,1,m,n,k,1,a,k,b,k,1,c,n);
}
#ifdef GPU
@ -173,6 +173,6 @@ void backward_connected_layer_gpu(connected_layer l, network_state state)
b = l.weights_gpu;
c = state.delta;
if(c) gemm_ongpu(0,1,m,n,k,1,a,k,b,k,0,c,n);
if(c) gemm_ongpu(0,1,m,n,k,1,a,k,b,k,1,c,n);
}
#endif