fix: multi-gpu training

sehoffmann · Mar 20, 2024 · 676044d · 676044d
1 parent 172550d
commit 676044d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dmlcloud/stage.py b/dmlcloud/stage.py
@@ -148,7 +148,8 @@ def _pre_stage(self):
         for handler in self.logger.handlers:
             handler.flush()
 
-        self.table._print_header()
+        if is_root():
+            self.table._print_header()
 
     def _post_stage(self):
         self.stop_time = datetime.now()