diff --git a/ultralytics/engine/validator.py b/ultralytics/engine/validator.py index 1f6f6912..5f5268ea 100644 --- a/ultralytics/engine/validator.py +++ b/ultralytics/engine/validator.py @@ -120,7 +120,7 @@ class BaseValidator: self.args.plots &= trainer.stopper.possible_stop or (trainer.epoch == trainer.epochs - 1) model.eval() else: - if str(self.args.model).endswith(".yaml"): + if str(self.args.model).endswith(".yaml") and model is None: LOGGER.warning("WARNING ⚠️ validating an untrained model YAML will result in 0 mAP.") callbacks.add_integration_callbacks(self) model = AutoBackend(