remove sync_dist due to overhead

drivendataorg · Nov 1, 2022 · 12ddfe5 · 12ddfe5
1 parent 8357e79
commit 12ddfe5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/zamba/pytorch_lightning/utils.py b/zamba/pytorch_lightning/utils.py
@@ -203,7 +203,6 @@ def compute_and_log_metrics(
         self.log(
             f"{subset}_macro_f1",
             f1_score(y_true, y_pred, average="macro", zero_division=0),
-            sync_dist=True,
         )
 
         # if only two classes, skip top_k accuracy since not enough classes
@@ -220,10 +219,9 @@ def compute_and_log_metrics(
                             labels=np.arange(y_proba.shape[1]),
                             k=k,
                         ),
-                        sync_dist=True,
                     )
         else:
-            self.log(f"{subset}_accuracy", accuracy_score(y_true, y_pred), sync_dist=True)
+            self.log(f"{subset}_accuracy", accuracy_score(y_true, y_pred))
 
         for metric_name, label, metric in compute_species_specific_metrics(
             y_true, y_pred, self.species