fix: double printing when training multiple times

sehoffmann · Nov 18, 2023 · 9e11fc6 · 9e11fc6
1 parent ea8e01f
commit 9e11fc6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dmlcloud/training/util.py b/dmlcloud/training/util.py
@@ -15,6 +15,9 @@ def setup_logging():
     root_logger = logging.getLogger()
     root_logger.setLevel(logging.INFO if hvd.rank() == 0 else logging.WARNING)
 
+    if root_logger.hasHandlers():
+        return
+
     stdout_handler = logging.StreamHandler(sys.stdout)
     stdout_handler.setLevel(logging.DEBUG)
     stdout_handler.setFormatter(logging.Formatter())