experimental change: sync_dist=True

NRCan · Nov 14, 2024 · 4a8fda1 · 4a8fda1
1 parent b58f69f
commit 4a8fda1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/geo_deep_learning/tasks_with_models/segmentation_dofa.py b/geo_deep_learning/tasks_with_models/segmentation_dofa.py
@@ -106,7 +106,7 @@ def test_step(self, batch, batch_idx):
                     break
     def on_test_epoch_end(self):
         print(f"test_metrics: {self.test_metrics}")
-        self.log_dict(self.test_metrics)
+        self.log_dict(self.test_metrics, sync_dist=True)
 
     def on_train_end(self):
         if self.trainer.is_global_zero and self.trainer.checkpoint_callback is not None: