From f70aa9e372606a6bd3c03bab9e322ccf58058348 Mon Sep 17 00:00:00 2001
From: rballester <rballeba@gmail.com>
Date: Fri, 16 Aug 2024 15:13:18 +0200
Subject: [PATCH] turning on sync for distributed setting

---
 code/models/base.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/code/models/base.py b/code/models/base.py
index 06fbf6f..ab81755 100644
--- 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):