Unverified 提交 5cb9fe61 authored 作者: Victor Sonck's avatar Victor Sonck 提交者: GitHub

Make sure best.pt model file is preserved ClearML (#9265)

* Make sure best.pt model file is preserved ClearML * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ciCo-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
上级 15e82d29
......@@ -233,7 +233,9 @@ class Loggers():
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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论