From 8bcdee86585f496afe1a8a38d608ea0504a11243 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 01 Sep 2015 18:22:03 +0000
Subject: [PATCH] Some bug fixes, random stuff
---
src/convolutional_layer.c | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index 7dcf5a4..6e3f38b 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -122,9 +122,9 @@
cuda_free(l->delta_gpu);
cuda_free(l->output_gpu);
- l->col_image_gpu = cuda_make_array(0, out_h*out_w*l->size*l->size*l->c);
- l->delta_gpu = cuda_make_array(0, l->batch*out_h*out_w*l->n);
- l->output_gpu = cuda_make_array(0, l->batch*out_h*out_w*l->n);
+ l->col_image_gpu = cuda_make_array(l->col_image, out_h*out_w*l->size*l->size*l->c);
+ l->delta_gpu = cuda_make_array(l->delta, l->batch*out_h*out_w*l->n);
+ l->output_gpu = cuda_make_array(l->output, l->batch*out_h*out_w*l->n);
#endif
}
@@ -261,7 +261,7 @@
int i;
for(i = 0; i < l.n; ++i){
filters[i] = copy_image(get_convolutional_filter(l, i));
- normalize_image(filters[i]);
+ //normalize_image(filters[i]);
}
return filters;
}
--
Gitblit v1.10.0