changes to log params

NRCan · Nov 18, 2024 · 6dc52e2 · 6dc52e2
1 parent 5180254
commit 6dc52e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/geo_deep_learning/tasks_with_models/segmentation_dofa.py b/geo_deep_learning/tasks_with_models/segmentation_dofa.py
@@ -106,8 +106,8 @@ def test_step(self, batch, batch_idx):
                     break
 
         self.log_dict(metrics,
-                      prog_bar=True, logger=True, 
-                      on_step=True, on_epoch=True, sync_dist=True, rank_zero_only=True)
+                      prog_bar=False, logger=True, 
+                      on_step=False, rank_zero_only=True)
 
     def on_train_end(self):
         if self.trainer.is_global_zero and self.trainer.checkpoint_callback is not None: