From 74aaab33129724e0f9f663cff268f7bb296c386b Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Tue, 12 Apr 2022 15:16:56 +0530 Subject: [PATCH] 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 --- utils/loggers/__init__.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index bab133cc35a9..3a3ec1ee455b 100644 --- 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