Refactor error handling in SegmentationCallback

NRCan · Nov 27, 2024 · e764b6b · e764b6b
1 parent a207b3f
commit e764b6b
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/geo_deep_learning/tools/callbacks/segmentation_callback.py b/geo_deep_learning/tools/callbacks/segmentation_callback.py
@@ -76,8 +76,4 @@ def _log_visualizations(self, trainer, batch_idx):
                                                         artifact_file = artifact_file,
                                                         run_id=trainer.logger.run_id)
             except Exception as e:
-                print(f"Error in logging visualizations: {e}")
-
-            finally:
-                self.current_batch = None
-                self.current_outputs = None
+                print(f"Error in logging visualizations: {e}")