OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Minor fix in cost_layer

This commit is contained in:
AlexeyAB 2019-06-05 01:04:16 +03:00
parent 1f6f2e1162
commit abba310982
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/cost_layer.c
View File

@ -51,8 +51,8 @@ cost_layer make_cost_layer(int batch, int inputs, COST_TYPE cost_type, float sca
l.forward_gpu = forward_cost_layer_gpu;
l.backward_gpu = backward_cost_layer_gpu;
l.delta_gpu = cuda_make_array(l.output, inputs*batch);
l.output_gpu = cuda_make_array(l.delta, inputs*batch);
l.delta_gpu = cuda_make_array(l.delta, inputs*batch);
l.output_gpu = cuda_make_array(l.output, inputs*batch);
#endif
return l;
}