Skip to content

Commit

Permalink
Merge pull request pjreddie#422 from Alexey-Kamenev/master
Browse files Browse the repository at this point in the history
Fixed YOLOv1 training.
  • Loading branch information
pjreddie committed Mar 25, 2018
2 parents 80d9bec + 4c0382d commit 8152938
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions src/detection_layer.c
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down

0 comments on commit 8152938

Please sign in to comment.