From fdcc096694791f92d2cff0e4fd74fbabfca12936 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 24 Jul 2015 22:14:23 +0000
Subject: [PATCH] Fixed some problems with visualizing networks

---
 src/convolutional_layer.c |    3 +--
 1 files changed, 1 insertions(+), 2 deletions(-)

diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index c266934..378e23f 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -188,8 +188,6 @@
     gradient_array(l.output, m*k*l.batch, l.activation, l.delta);
     backward_bias(l.bias_updates, l.delta, l.batch, l.n, k);
 
-    if(state.delta) memset(state.delta, 0, l.batch*l.h*l.w*l.c*sizeof(float));
-
     for(i = 0; i < l.batch; ++i){
         float *a = l.delta + i*m*k;
         float *b = l.col_image;
@@ -250,6 +248,7 @@
     int i;
     for(i = 0; i < l.n; ++i){
         filters[i] = copy_image(get_convolutional_filter(l, i));
+        normalize_image(filters[i]);
     }
     return filters;
 }

--
Gitblit v1.10.0