From ae4ec1a06d5b4c9bcb5647cf555aafd4e5f800f7 Mon Sep 17 00:00:00 2001
From: Bartek Gąsiorzewski <g.bartek@gmail.com>
Date: Mon, 07 May 2018 20:11:57 +0000
Subject: [PATCH] Exit with nonzero status on error

---
 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