From 51fbf0127726a659862c9ee3b6b7257fc0a1b39d Mon Sep 17 00:00:00 2001
From: Philip Kahn <tigerhawkvok@gmail.com>
Date: Fri, 04 May 2018 22:07:45 +0000
Subject: [PATCH] Address GPU message in root dir

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