From c1bb8c129d7d4d8fc85348d5e7ac5441fa36ead5 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sat, 19 May 2018 13:52:05 +0000
Subject: [PATCH] Fixed xnor for random=1
---
src/convolutional_layer.c | 9 +++++++++
1 files changed, 9 insertions(+), 0 deletions(-)
diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index a349454..f4a5d89 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -465,6 +465,10 @@
l->x_norm = realloc(l->x_norm, l->batch*l->outputs*sizeof(float));
}
+ if (l->xnor) {
+ //l->binary_input = realloc(l->inputs*l->batch, sizeof(float));
+ }
+
#ifdef GPU
if (old_w < w || old_h < h) {
cuda_free(l->delta_gpu);
@@ -480,6 +484,11 @@
l->x_gpu = cuda_make_array(l->output, l->batch*l->outputs);
l->x_norm_gpu = cuda_make_array(l->output, l->batch*l->outputs);
}
+
+ if (l->xnor) {
+ cuda_free(l->binary_input_gpu);
+ l->binary_input_gpu = cuda_make_array(0, l->inputs*l->batch);
+ }
}
#ifdef CUDNN
cudnn_convolutional_setup(l, cudnn_fastest);
--
Gitblit v1.10.0