Fix validator model=None (#16699)

2 months ago · 2e084c0581
parent 4b95ae70a0
commit 2e084c0581
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/engine/validator.py
+++ b/ultralytics/engine/validator.py
@ -119,7 +119,7 @@ class BaseValidator:
            self.args.plots &= trainer.stopper.possible_stop or (trainer.epoch == trainer.epochs - 1)
            model.eval()
        else:
-            if Path(self.args.model).suffix == ".yaml":
+            if str(self.args.model).endswith(".yaml"):
                LOGGER.warning("WARNING ⚠️ validating an untrained model YAML will result in 0 mAP.")
            callbacks.add_integration_callbacks(self)
            model = AutoBackend(