sync dist log

discovery-unicamp · May 14, 2024 · 77c0828 · 77c0828
1 parent 3ace8d2
commit 77c0828
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/minerva/models/nets/setr.py b/minerva/models/nets/setr.py
@@ -601,6 +601,7 @@ def _single_step(self, batch: torch.Tensor, batch_idx: int, step_name: str):
                 on_epoch=True,
                 prog_bar=True,
                 logger=True,
+                sync_dist=True,
             )
 
         return loss