Fix validator model=None (#16699)

This commit is contained in:
Glenn Jocher 2024-10-05 00:53:19 +02:00 committed by GitHub
parent 4b95ae70a0
commit 2e084c0581
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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(