From f451cc936991e5095b1cb02262b5b0a9f12693f1 Mon Sep 17 00:00:00 2001
From: jvech <jmvalenciae@unal.edu.co>
Date: Sun, 8 Sep 2024 09:25:50 -0500
Subject: fix: memory leaks fixed

---
 src/main.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/main.c')

diff --git a/src/main.c b/src/main.c
index 63a2fa9..01ada0a 100644
--- a/src/main.c
+++ b/src/main.c
@@ -152,6 +152,8 @@ int main(int argc, char *argv[]) {
 
     nn_network_free_weights(network, ml_configs.network_size);
     free(network);
+    free(X.data);
+    free(y.data);
     util_free_config(&ml_configs);
     return 0;
 }
-- 
cgit v1.2.3-70-g09d2