From b18fd7aab9cb3059c223f574062c84d8fcd69b94 Mon Sep 17 00:00:00 2001
From: augustin64 <me.git@augustin64.fr>
Date: Sat, 4 Feb 2023 13:12:52 +0100
Subject: [PATCH] Update headers

---
 src/cnn/include/main.h  | 4 ++--
 src/cnn/include/train.h | 9 +++++++--
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/cnn/include/main.h b/src/cnn/include/main.h
index 3d29d1c..a5ab748 100644
--- a/src/cnn/include/main.h
+++ b/src/cnn/include/main.h
@@ -7,8 +7,8 @@
 void help(char* call);
 
 /*
- * Fonction utilisée à des fins de développement du réseau convolutif
+ * Fonction principale
 */
-void dev_conv();
+int main(int argc, char* argv[]);
 
 #endif
\ No newline at end of file
diff --git a/src/cnn/include/train.h b/src/cnn/include/train.h
index 7913060..34e139e 100644
--- a/src/cnn/include/train.h
+++ b/src/cnn/include/train.h
@@ -6,7 +6,7 @@
 
 #define EPOCHS 10
 #define BATCHES 500
-//#define USE_MULTITHREADING
+#define USE_MULTITHREADING
 #define LEARNING_RATE 0.05
 
 
@@ -23,11 +23,16 @@ typedef struct TrainParameters {
     int height; // Hauteur des images
     int dataset_type; // Type de dataset
     int start; // Début des images
-    int nb_images; // Nombre d'images àn traiter
+    int nb_images; // Nombre d'images à traiter
     float accuracy; // Accuracy (à renvoyer)
     float loss; // Loss (à renvoyer)
 } TrainParameters;
 
+/*
+ * Partie entière supérieure de a/b
+*/
+int div_up(int a, int b);
+
 
 /*
  * Fonction auxiliaire d'entraînement destinée à être exécutée sur plusieurs threads à la fois