this fixes the issue

choderalab · Nov 7, 2024 · 667909f · 667909f
1 parent 8b23f18
commit 667909f
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/modelforge/train/training.py b/modelforge/train/training.py
@@ -1157,9 +1157,7 @@ def _log_figures_for_each_phase(
             # Log outlier error counts for non-training phases
             if phase != "train":
                 self._identify__and_log_top_k_errors(errors, gathered_indices, phase)
-                self.log_dict(
-                    self.outlier_errors_over_epochs, on_epoch=True, sync_dist=True
-                )
+                self.log_dict(self.outlier_errors_over_epochs, on_epoch=True)
 
     def _identify__and_log_top_k_errors(
         self,
@@ -1299,7 +1297,6 @@ def on_train_epoch_end(self):
             self.log(
                 f"loss/{key}/weight",
                 weight[self.current_epoch],
-                sync_dist=True,
             )
 
     def _log_learning_rate(self):
@@ -1311,7 +1308,6 @@ def _log_learning_rate(self):
                 sch.get_last_lr()[0],
                 on_epoch=True,
                 prog_bar=True,
-                sync_dist=True,
             )
         except AttributeError:
             pass