diff --git a/src/detection_layer.c b/src/detection_layer.c index 5c8a1cea..a2a3c842 100644 --- a/src/detection_layer.c +++ b/src/detection_layer.c @@ -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);