Add version warning for wandb (#7385)

* add version warning * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update __init__.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <[email protected]>
zhongcaoren95 · Apr 12, 2022 · 74aaab3 · 74aaab3
1 parent 4bb7eb8
commit 74aaab3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py
@@ -81,6 +81,11 @@ def __init__(self, save_dir=None, weights=None, opt=None, hyp=None, logger=None,
             run_id = torch.load(self.weights).get('wandb_id') if self.opt.resume and not wandb_artifact_resume else None
             self.opt.hyp = self.hyp  # add hyperparameters
             self.wandb = WandbLogger(self.opt, run_id)
+            # temp warn. because nested artifacts not supported after 0.12.10
+            if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.11'):
+                self.logger.warning(
+                    "YOLOv5 temporarily requires wandb version 0.12.10 or below. Some features may not work as expected."
+                )
         else:
             self.wandb = None