ultralytics 8.0.198 MLflow fix, tests and Docs page (#5357)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-10-13 20:41:05 +02:00 · 2023-10-13 20:41:05 +02:00 · 5b3c4cfc0e
commit 5b3c4cfc0e
parent 12e3eef844
11 changed files with 228 additions and 65 deletions
--- a/ultralytics/engine/trainer.py
+++ b/ultralytics/engine/trainer.py
@ -91,6 +91,7 @@ class BaseTrainer:

        # Dirs
        self.save_dir = get_save_dir(self.args)
+        self.args.name = self.save_dir.name  # update name for loggers
        self.wdir = self.save_dir / 'weights'  # weights dir
        if RANK in (-1, 0):
            self.wdir.mkdir(parents=True, exist_ok=True)  # make dir