From ab75d5c5783db4792e400f933d584984f3aa7bf0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 20 Jun 2016 05:39:37 +0000
Subject: [PATCH] t1
---
src/network.c | 8 +++++++-
1 files changed, 7 insertions(+), 1 deletions(-)
diff --git a/src/network.c b/src/network.c
index 88b7085..51f74d9 100644
--- a/src/network.c
+++ b/src/network.c
@@ -253,6 +253,7 @@
int i;
float *original_input = state.input;
float *original_delta = state.delta;
+ state.workspace = net.workspace;
for(i = net.n-1; i >= 0; --i){
state.index = i;
if(i == 0){
@@ -391,6 +392,11 @@
int i;
for(i = 0; i < net->n; ++i){
net->layers[i].batch = b;
+ #ifdef CUDNN
+ if(net->layers[i].type == CONVOLUTIONAL){
+ cudnn_convolutional_setup(net->layers + i);
+ }
+ #endif
}
}
@@ -433,7 +439,7 @@
net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1);
#else
free(net->workspace);
- net->workspace = calloc(1, (workspace_size-1)/sizeof(float)+1);
+ net->workspace = calloc(1, workspace_size);
#endif
//fprintf(stderr, " Done!\n");
return 0;
--
Gitblit v1.10.0