Ayush Chaurasia pre-commit-ci[bot] glenn-jocher commited on
Commit
74aaab3
·
unverified ·
1 Parent(s): 4bb7eb8

Add version warning for wandb (#7385)

Browse files

* 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]>

Files changed (1) hide show
  1. utils/loggers/__init__.py +5 -0
utils/loggers/__init__.py CHANGED
@@ -81,6 +81,11 @@ class Loggers():
81
  run_id = torch.load(self.weights).get('wandb_id') if self.opt.resume and not wandb_artifact_resume else None
82
  self.opt.hyp = self.hyp # add hyperparameters
83
  self.wandb = WandbLogger(self.opt, run_id)
 
 
 
 
 
84
  else:
85
  self.wandb = None
86
 
 
81
  run_id = torch.load(self.weights).get('wandb_id') if self.opt.resume and not wandb_artifact_resume else None
82
  self.opt.hyp = self.hyp # add hyperparameters
83
  self.wandb = WandbLogger(self.opt, run_id)
84
+ # temp warn. because nested artifacts not supported after 0.12.10
85
+ if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.11'):
86
+ self.logger.warning(
87
+ "YOLOv5 temporarily requires wandb version 0.12.10 or below. Some features may not work as expected."
88
+ )
89
  else:
90
  self.wandb = None
91