From fe6377ec68756c8cbad56a508d3001a34f2e6670 Mon Sep 17 00:00:00 2001
From: Achille Mbogol Touye <achille.mbogol-touye@univ-grenoble-alpes.fr>
Date: Tue, 7 Nov 2023 01:47:47 +0100
Subject: [PATCH] Nice progressbar
---
Wine.Lightning/modules/progressbar.py | 37 +++++++++++++++++++++++++++
1 file changed, 37 insertions(+)
create mode 100644 Wine.Lightning/modules/progressbar.py
diff --git a/Wine.Lightning/modules/progressbar.py b/Wine.Lightning/modules/progressbar.py
new file mode 100644
index 0000000..075e13b
--- /dev/null
+++ b/Wine.Lightning/modules/progressbar.py
@@ -0,0 +1,37 @@
+# ------------------------------------------------------------------
+# _____ _ _ _
+# | ___(_) __| | | ___
+# | |_ | |/ _` | |/ _ \
+# | _| | | (_| | | __/
+# |_| |_|\__,_|_|\___|
+# ------------------------------------------------------------------
+# Formation Introduction au Deep Learning (FIDLE)
+# CNRS/SARI/DEVLOG 2023
+# ------------------------------------------------------------------
+# 2.0 version by Achille Mbogol Touye (EFELIA-MIAI/SIMAP¨), sep 2023
+
+from tqdm import tqdm as _tqdm
+from lightning.pytorch.callbacks import TQDMProgressBar
+
+# Créez un callback de barre de progression pour afficher les métriques d'entraînement
+class CustomTrainProgressBar(TQDMProgressBar):
+ def __init__(self):
+ super().__init__()
+ self._val_progress_bar = _tqdm()
+
+ def init_train_tqdm(self):
+ bar=super().init_train_tqdm()
+ bar.set_description("Training")
+ return bar
+
+ @property
+ def val_progress_bar(self):
+ if self._val_progress_bar is None:
+ raise ValueError("The `_val_progress_bar` reference has not been set yet.")
+ return self._val_progress_bar
+
+
+ def on_validation_start(self, trainer, pl_module):
+ # Désactivez l'affichage de la barre de progression de validation
+ self.val_progress_bar.disable = True
+
\ No newline at end of file
--
GitLab