fix: memory leaks fixed

1 files changed, 3 insertions, 0 deletions

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");