From daaef0971b8a34e25e6eab1637c336deda11e500 Mon Sep 17 00:00:00 2001
From: augustin64 <augustin.lucas64@gmail.com>
Date: Tue, 26 Apr 2022 17:29:25 +0200
Subject: [PATCH] Update main.c

---
 src/mnist/main.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mnist/main.c b/src/mnist/main.c
index 0206cba..a602c4f 100644
--- a/src/mnist/main.c
+++ b/src/mnist/main.c
@@ -222,7 +222,7 @@ void test(char* modele, char* fichier_images, char* fichier_labels, bool preview
     for (int i=0; i < nb_images; i++) {
         if (indice_max(resultats[i], nb_der_layer) == labels[i]) {
             accuracy += 1. / (float)nb_images;
-        } else {
+        } else if (preview_fails) {
             printf("--- Image %d, %d --- Prévision: %d ---\n", i, labels[i], indice_max(resultats[i], nb_der_layer));
             print_image(width, height, images[i], resultats[i]);
         }