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:
parent
87233ea17c
commit
9a5891444e
48 changed files with 95 additions and 92 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue