From 892923514f66d9b6acefd44f3ddf688bd7e6f268 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 08 Dec 2015 23:12:10 +0000
Subject: [PATCH] fixed darknet, stuff
---
src/connected_layer.c | 13 ++++++++++---
1 files changed, 10 insertions(+), 3 deletions(-)
diff --git a/src/connected_layer.c b/src/connected_layer.c
index bff3602..640e8b8 100644
--- a/src/connected_layer.c
+++ b/src/connected_layer.c
@@ -29,7 +29,8 @@
l.biases = calloc(outputs, sizeof(float));
- float scale = 1./sqrt(inputs);
+ //float scale = 1./sqrt(inputs);
+ float scale = sqrt(2./inputs);
for(i = 0; i < inputs*outputs; ++i){
l.weights[i] = 2*scale*rand_uniform() - scale;
}
@@ -102,7 +103,7 @@
b = l.weights;
c = state.delta;
- if(c) gemm(0,1,m,n,k,1,a,k,b,k,0,c,n);
+ if(c) gemm(0,1,m,n,k,1,a,k,b,k,1,c,n);
}
#ifdef GPU
@@ -147,6 +148,12 @@
float * c = l.output_gpu;
gemm_ongpu(0,0,m,n,k,1,a,k,b,n,1,c,n);
activate_array_ongpu(l.output_gpu, l.outputs*l.batch, l.activation);
+
+/*
+ cuda_pull_array(l.output_gpu, l.output, l.outputs*l.batch);
+ float avg = mean_array(l.output, l.outputs*l.batch);
+ printf("%f\n", avg);
+ */
}
void backward_connected_layer_gpu(connected_layer l, network_state state)
@@ -172,6 +179,6 @@
b = l.weights_gpu;
c = state.delta;
- if(c) gemm_ongpu(0,1,m,n,k,1,a,k,b,k,0,c,n);
+ if(c) gemm_ongpu(0,1,m,n,k,1,a,k,b,k,1,c,n);
}
#endif
--
Gitblit v1.10.0