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

Merge pull request #422 from Alexey-Kamenev/master

Browse Source 
Fixed YOLOv1 training.
This commit is contained in:
Joseph Redmon 2018-03-24 18:56:36 -07:00 committed by GitHub
commit 815293875c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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;
}
//float *in_cpu = calloc(l.batch*l.inputs, sizeof(float));
//float *truth_cpu = 0;
cuda_pull_array(net.input_gpu, net.input, l.batch*l.inputs);
forward_detection_layer(l, net);
cuda_push_array(l.output_gpu, l.output, l.batch*l.outputs);
cuda_push_array(l.delta_gpu, l.delta, l.batch*l.inputs);