From ac46d97351fdaf55f7d44a1a6fbba8e5951b12f9 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 14 Mar 2016 06:25:36 +0000
Subject: [PATCH] less printing
---
src/route_layer.c | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/route_layer.c b/src/route_layer.c
index 67b606c..df50b64 100644
--- a/src/route_layer.c
+++ b/src/route_layer.c
@@ -21,11 +21,11 @@
fprintf(stderr, "\n");
l.outputs = outputs;
l.inputs = outputs;
- l.delta = calloc(outputs*batch, sizeof(float));
+ l.delta = calloc(outputs*batch, sizeof(float));
l.output = calloc(outputs*batch, sizeof(float));;
#ifdef GPU
- l.delta_gpu = cuda_make_array(0, outputs*batch);
- l.output_gpu = cuda_make_array(0, outputs*batch);
+ l.delta_gpu = cuda_make_array(l.delta, outputs*batch);
+ l.output_gpu = cuda_make_array(l.output, outputs*batch);
#endif
return l;
}
--
Gitblit v1.10.0