ultralytics 8.0.209 fix P6 model validation (#6261)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Jie Li <32835610+jedi007@users.noreply.github.com>
Co-authored-by: Haruto Watahiki <43723360+WATA-Haru@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-11-14 00:54:30 +01:00 committed by GitHub
parent 87233ea17c
commit 9a5891444e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
48 changed files with 95 additions and 92 deletions

View file

@ -77,7 +77,7 @@ class BaseValidator:
self.args = get_cfg(overrides=args)
self.dataloader = dataloader
self.pbar = pbar
self.model = None
self.stride = None
self.data = None
self.device = None
self.batch_i = None
@ -146,6 +146,7 @@ class BaseValidator:
self.args.workers = 0 # faster CPU val as time dominated by inference, not dataloading
if not pt:
self.args.rect = False
self.stride = model.stride # used in get_dataloader() for padding
self.dataloader = self.dataloader or self.get_dataloader(self.data.get(self.args.split), self.args.batch)
model.eval()