fix: memory leaks fixed

2 files changed, 5 insertions, 0 deletions

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;
 }
diff --git a/src/nn.c b/src/nn.c
index 19076bb..e905bf3 100644
--- a/src/nn.c
+++ b/src/nn.c
@@ -170,6 +170,9 @@ void nn_network_train(
     free(weights);
     free(biases);
 
+    free(input_random);
+    free(labels_random);
+
     return;
 nn_network_train_error:
     perror("nn_network_train() Error");