From 8bcdee86585f496afe1a8a38d608ea0504a11243 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 01 Sep 2015 18:22:03 +0000
Subject: [PATCH] Some bug fixes, random stuff
---
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