From 3ebcc647b651a4a3c717eff2a3087127e5707e0c Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 14 May 2018 09:20:38 +0000
Subject: [PATCH] Fixed network resizing (random=1) for non-square networks

---
 src/network.c |    5 +++++
 1 files changed, 5 insertions(+), 0 deletions(-)

diff --git a/src/network.c b/src/network.c
index ce6d28a..81b53f3 100644
--- a/src/network.c
+++ b/src/network.c
@@ -767,6 +767,11 @@
 		free_layer(net.layers[i]);
 	}
 	free(net.layers);
+
+	free(net.scales);
+	free(net.steps);
+	free(net.seen);
+
 #ifdef GPU
 	if (gpu_index >= 0) cuda_free(net.workspace);
 	else free(net.workspace);

--
Gitblit v1.10.0