Zero mAP warning on YAML val (#16698)
This commit is contained in:
parent
c6371ba8c4
commit
e5d3427a52
1 changed files with 2 additions and 0 deletions
|
|
@ -119,6 +119,8 @@ 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":
|
||||
LOGGER.warning("WARNING ⚠️ validating an untrained model YAML will result in 0 mAP.")
|
||||
callbacks.add_integration_callbacks(self)
|
||||
model = AutoBackend(
|
||||
weights=model or self.args.model,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue