Integrate OpenVINO CUMULATIVE_THROUGHPUT mode batched inference (#8834)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-11 18:50:29 +01:00 · 2024-03-11 18:50:29 +01:00 · 7638c5ce4d
commit 7638c5ce4d
parent ff8ae0d2e2
3 changed files with 9 additions and 4 deletions
--- a/ultralytics/engine/validator.py
+++ b/ultralytics/engine/validator.py
@ -122,7 +122,7 @@ class BaseValidator:
        else:
            callbacks.add_integration_callbacks(self)
            model = AutoBackend(
-                model or self.args.model,
+                weights=model or self.args.model,
                device=select_device(self.args.device, self.args.batch),
                dnn=self.args.dnn,
                data=self.args.data,