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 | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/connected_layer.c b/src/connected_layer.c
index 55d84ca..4323505 100644
--- a/src/connected_layer.c
+++ b/src/connected_layer.c
@@ -103,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
@@ -173,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