From ec68838342b42541776607b0c14e40fb89f7e3d8 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 23 May 2018 15:27:18 +0000
Subject: [PATCH] Fixed memory leaks for Yolo: train, test

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

diff --git a/src/layer.c b/src/layer.c
index 582cbb3..3b47917 100644
--- a/src/layer.c
+++ b/src/layer.c
@@ -11,6 +11,7 @@
 #endif
 		return;
 	}
+	if (l.mask)               free(l.mask);
 	if (l.cweights)           free(l.cweights);
 	if (l.indexes)            free(l.indexes);
 	if (l.input_layers)       free(l.input_layers);

--
Gitblit v1.10.0