Update trainer.py

luxonis · Feb 28, 2024 · 986680b · 986680b
1 parent 69bd044
commit 986680b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/luxonis_train/core/trainer.py b/luxonis_train/core/trainer.py
@@ -43,12 +43,12 @@ def __init__(
     def _upload_logs(self) -> None:
         if self.cfg.tracker.is_mlflow:
             logger.info("Uploading logs to MLFlow.")
-            self.fs = LuxonisFileSystem(
+            fs = LuxonisFileSystem(
                 "mlflow://",
                 allow_active_mlflow_run=True,
                 allow_local=False,
             )
-            self.fs.put_file(
+            fs.put_file(
                 local_path=self.log_file,
                 remote_path="luxonis_train.log",
                 mlflow_instance=self.tracker.experiment.get("mlflow", None),