From 153705226d8ca746478b69eeac9bc854766daa11 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 27 Jan 2015 21:31:06 +0000
Subject: [PATCH] Bias updates bug fix
---
src/network.c | 17 +++++++++--------
1 files changed, 9 insertions(+), 8 deletions(-)
diff --git a/src/network.c b/src/network.c
index 641d782..f554090 100644
--- a/src/network.c
+++ b/src/network.c
@@ -42,8 +42,6 @@
return "none";
}
-
-
network make_network(int n, int batch)
{
network net;
@@ -53,14 +51,14 @@
net.types = calloc(net.n, sizeof(LAYER_TYPE));
net.outputs = 0;
net.output = 0;
+ net.seen = 0;
#ifdef GPU
- net.input_cl = calloc(1, sizeof(cl_mem));
- net.truth_cl = calloc(1, sizeof(cl_mem));
+ net.input_gpu = calloc(1, sizeof(float *));
+ net.truth_gpu = calloc(1, sizeof(float *));
#endif
return net;
}
-
void forward_network(network net, float *input, float *truth, int train)
{
int i;
@@ -107,9 +105,12 @@
}
else if(net.types[i] == FREEWEIGHT){
if(!train) continue;
- freeweight_layer layer = *(freeweight_layer *)net.layers[i];
- forward_freeweight_layer(layer, input);
+ //freeweight_layer layer = *(freeweight_layer *)net.layers[i];
+ //forward_freeweight_layer(layer, input);
}
+ //char buff[256];
+ //sprintf(buff, "layer %d", i);
+ //cuda_compare(get_network_output_gpu_layer(net, i), input, get_network_output_size_layer(net, i)*net.batch, buff);
}
}
@@ -582,7 +583,7 @@
float *network_predict(network net, float *input)
{
#ifdef GPU
- if(gpu_index >= 0) return network_predict_gpu(net, input);
+ if(gpu_index >= 0) return network_predict_gpu(net, input);
#endif
forward_network(net, input, 0, 0);
--
Gitblit v1.10.0