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 {