From 37259078f02b797ece6bf67aa6179cb05ba92d47 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sat, 05 Aug 2017 13:23:53 +0000
Subject: [PATCH] Add tracking for image in example of usage SO/DLL

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

diff --git a/src/network.c b/src/network.c
index 0a49bf1..79940e6 100644
--- a/src/network.c
+++ b/src/network.c
@@ -595,11 +595,14 @@
 		free_layer(net.layers[i]);
 	}
 	free(net.layers);
-	free(net.workspace);
 #ifdef GPU
+	if (gpu_index >= 0) cuda_free(net.workspace);
+	else free(net.workspace);
 	if (*net.input_gpu) cuda_free(*net.input_gpu);
 	if (*net.truth_gpu) cuda_free(*net.truth_gpu);
 	if (net.input_gpu) free(net.input_gpu);
 	if (net.truth_gpu) free(net.truth_gpu);
+#else
+	free(net.workspace);
 #endif
 }

--
Gitblit v1.10.0