turning on sync for distributed setting

aidos-lab · Aug 16, 2024 · f70aa9e · f70aa9e
1 parent c9ec267
commit f70aa9e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/code/models/base.py b/code/models/base.py
@@ -94,6 +94,7 @@ def log_accuracies(
             x_hat,
             y,
             step,
+            sync_dist=True
         )
         for accuracy in accuracies:
             self.log(
@@ -102,6 +103,7 @@ def log_accuracies(
                 on_step=False,
                 on_epoch=True,
                 batch_size=batch_len,
+                sync_dist=True
             )
 
     def test_step(self, batch, batch_idx):