From d0b9326a352ed2fbc3ae66fdef40b4533a2f211d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 11 Aug 2015 06:22:27 +0000
Subject: [PATCH] Hacks to get nightmare to not break gridsizing
---
src/connected_layer.c | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/connected_layer.c b/src/connected_layer.c
index bff3602..4323505 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
@@ -172,6 +173,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