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

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

diff --git a/code/models/base.py b/code/models/base.py
index 57d4bd2..8610530 100644
--- a/code/models/base.py
+++ b/code/models/base.py
@@ -67,6 +67,7 @@ def general_step(self, batch, batch_idx, step: str):
             batch_size=batch_len,
             on_step=False,
             on_epoch=True,
+            sync_dist=True
         )
         self.log_accuracies(x_hat, batch.y, batch_len, step)
         return loss