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

Fixed YOLOv1 training.

Browse Source
This commit is contained in:
Alexey Kamenev 2018-01-18 13:50:47 -08:00
parent 80d9bec20f
commit 4c0382d423
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/detection_layer.c
View File

@ -260,9 +260,7 @@ void forward_detection_layer_gpu(const detection_layer l, network net)
return; return;
} }
//float *in_cpu = calloc(l.batch*l.inputs, sizeof(float)); cuda_pull_array(net.input_gpu, net.input, l.batch*l.inputs);
//float *truth_cpu = 0;
forward_detection_layer(l, net); forward_detection_layer(l, net);
cuda_push_array(l.output_gpu, l.output, l.batch*l.outputs); cuda_push_array(l.output_gpu, l.output, l.batch*l.outputs);
cuda_push_array(l.delta_gpu, l.delta, l.batch*l.inputs); cuda_push_array(l.delta_gpu, l.delta, l.batch*l.inputs);