From 4c0382d4232a4b1a5e1a177aea26dbbeb17a5107 Mon Sep 17 00:00:00 2001
From: Alexey Kamenev <akamenev@nvidia.com>
Date: Thu, 18 Jan 2018 13:50:47 -0800
Subject: [PATCH] Fixed YOLOv1 training.

---
 src/detection_layer.c | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

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);