add method to remove file handlers keeping the log file open

luxonis · Mar 18, 2024 · adf5f65 · adf5f65
1 parent 331eae7
commit adf5f65
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/luxonis_train/core/core.py b/luxonis_train/core/core.py
@@ -234,3 +234,7 @@ def get_best_metric_checkpoint_path(self) -> str:
         @return: Path to best checkpoint with respect to best validation metric
         """
         return self.pl_trainer.checkpoint_callbacks[1].best_model_path  # type: ignore
+
+    def reset_logging(self) -> None:
+        """Close file handlers to release the log file."""
+        reset_logging()