From 8535d36885103c2089f6927301652ad6fcebc0d5 Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Thu, 12 Dec 2024 04:06:06 +0800 Subject: [PATCH] Fix untrained warning when training from yaml (#18168) Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Co-authored-by: Glenn Jocher --- ultralytics/engine/validator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(