From 08993ade85732487a106415df75c681b765960e6 Mon Sep 17 00:00:00 2001
From: augustin64 <me.git@augustin64.fr>
Date: Fri, 19 May 2023 23:29:46 +0200
Subject: [PATCH] cnn/train: add newline & change initialisation

---
 src/cnn/train.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/cnn/train.c b/src/cnn/train.c
index 248284b..0d7f6d3 100644
--- a/src/cnn/train.c
+++ b/src/cnn/train.c
@@ -234,10 +234,10 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
             network = create_network_lenet5(LEARNING_RATE, 0, LEAKY_RELU, HE, input_width, input_depth);
             //network = create_simple_one(LEARNING_RATE, 0, RELU, GLOROT, input_width, input_depth);    
         } else {
-            network = create_network_VGG16(LEARNING_RATE, 0, RELU, NORMALIZED_XAVIER, dataset->numCategories);
+            network = create_network_VGG16(LEARNING_RATE, 0, RELU, HE, dataset->numCategories);
 
             #ifdef USE_MULTITHREADING
-                printf_warning("Utilisation de VGG16 avec multithreading. La quantité de RAM utilisée peut devenir excessive");
+                printf_warning("Utilisation de VGG16 avec multithreading. La quantité de RAM utilisée peut devenir excessive\n");
             #endif
         }
     } else {