Fix validator model=None (#16699)
This commit is contained in:
parent
4b95ae70a0
commit
2e084c0581
1 changed files with 1 additions and 1 deletions
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue