From 2fab8e8f99d5e2936047d6fdb16092c9ae26d88f Mon Sep 17 00:00:00 2001 From: Victor Sonck Date: Fri, 2 Sep 2022 21:27:25 +0200 Subject: [PATCH 1/2] Make sure best.pt model file is preserved ClearML --- utils/loggers/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 1aa8427f9127..c981a0945642 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -233,7 +233,11 @@ def on_train_end(self, last, best, epoch, results): self.wandb.finish_run() if self.clearml and not self.opt.evolve: - self.clearml.task.update_output_model(model_path=str(best if best.exists() else last), name='Best Model') + self.clearml.task.update_output_model( + model_path=str(best if best.exists() else last), + name='Best Model', + auto_delete_file=False + ) def on_params_update(self, params: dict): # Update hyperparams or configs of the experiment From 4f6f200379734868333c6b1c80b46243e30bf061 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 2 Sep 2022 19:29:15 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- utils/loggers/__init__.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index c981a0945642..3aee35844f52 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -233,11 +233,9 @@ def on_train_end(self, last, best, epoch, results): self.wandb.finish_run() if self.clearml and not self.opt.evolve: - self.clearml.task.update_output_model( - model_path=str(best if best.exists() else last), - name='Best Model', - auto_delete_file=False - ) + self.clearml.task.update_output_model(model_path=str(best if best.exists() else last), + name='Best Model', + auto_delete_file=False) def on_params_update(self, params: dict): # Update hyperparams or configs of the experiment