turning on sync for distributed setting

aidos-lab · Aug 16, 2024 · b070e88 · b070e88
1 parent f70aa9e
commit b070e88
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/code/models/base.py b/code/models/base.py
@@ -93,8 +93,7 @@ def log_accuracies(
             acc_fun,
             x_hat,
             y,
-            step,
-            sync_dist=True
+            step
         )
         for accuracy in accuracies:
             self.log(