From eaf033c0570308dfcd381ed61d274c7f5add7cfc Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 09 Nov 2015 21:27:02 +0000
Subject: [PATCH] Added tiny yolo model
---
src/route_layer.c | 10 +++++-----
1 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/route_layer.c b/src/route_layer.c
index e3802b7..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;
}
@@ -54,7 +54,7 @@
float *delta = net.layers[index].delta;
int input_size = l.input_sizes[i];
for(j = 0; j < l.batch; ++j){
- copy_cpu(input_size, l.delta + offset + j*l.outputs, 1, delta + j*input_size, 1);
+ axpy_cpu(input_size, 1, l.delta + offset + j*l.outputs, 1, delta + j*input_size, 1);
}
offset += input_size;
}
@@ -85,7 +85,7 @@
float *delta = net.layers[index].delta_gpu;
int input_size = l.input_sizes[i];
for(j = 0; j < l.batch; ++j){
- copy_ongpu(input_size, l.delta_gpu + offset + j*l.outputs, 1, delta + j*input_size, 1);
+ axpy_ongpu(input_size, 1, l.delta_gpu + offset + j*l.outputs, 1, delta + j*input_size, 1);
}
offset += input_size;
}
--
Gitblit v1.10.0