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