From 2774cd86d41df2f4a5cf576a72418cadf90b4072 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 26 Nov 2015 22:40:49 +0000
Subject: [PATCH] threshs are floats
---
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