From 2790883586ccbd39431c698b51780f3f2c05d6d5 Mon Sep 17 00:00:00 2001
From: julienChemillier <chemillier.julien@gmail.com>
Date: Sat, 13 May 2023 10:05:54 +0200
Subject: [PATCH] Removal of warnings

---
 src/cnn/train.c | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/src/cnn/train.c b/src/cnn/train.c
index 3a8c068..0197659 100644
--- a/src/cnn/train.c
+++ b/src/cnn/train.c
@@ -63,9 +63,8 @@ void* train_thread(void* parameters) {
     float loss = 0.;
 
     pthread_t tid;
-    LoadImageParameters* load_image_param;
+    LoadImageParameters* load_image_param = (LoadImageParameters*)malloc(sizeof(LoadImageParameters));
     if (dataset_type != 0) {
-        load_image_param = (LoadImageParameters*)malloc(sizeof(LoadImageParameters));
         load_image_param->dataset = param->dataset;
         load_image_param->index = index[start];
 
@@ -118,9 +117,7 @@ void* train_thread(void* parameters) {
         }
     }
 
-    if (dataset_type != 0) {
-        free(load_image_param);
-    }
+    free(load_image_param);
 
     param->accuracy = accuracy;
     param->loss = loss;
@@ -140,6 +137,7 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
     float loss;
     float batch_loss; // May be redundant with loss, but gives more informations
     float test_accuracy = 0.; // Used to decrease Learning rate
+    (void)test_accuracy; // To avoid warnings when not used
     float accuracy;
     float batch_accuracy;
     float current_accuracy;