From 4ab366a805a7678642539465d68ef906b4599aeb Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 22 Dec 2014 22:35:37 +0000
Subject: [PATCH] some fixes, some other experiments

---
 src/network_gpu.c |    1 +
 1 files changed, 1 insertions(+), 0 deletions(-)

diff --git a/src/network_gpu.c b/src/network_gpu.c
index 6ff95c8..4d2c8d3 100644
--- a/src/network_gpu.c
+++ b/src/network_gpu.c
@@ -195,6 +195,7 @@
     }
     else if(net.types[i] == CONNECTED){
         connected_layer layer = *(connected_layer *)net.layers[i];
+        cl_read_array(layer.output_cl, layer.output, layer.outputs*layer.batch);
         return layer.output;
     }
     else if(net.types[i] == MAXPOOL){

--
Gitblit v1.10.0