From ee5331bdeee065fe74793bc18d7d0a522f076c38 Mon Sep 17 00:00:00 2001 From: Rick-v-E <78429257+Rick-v-E@users.noreply.github.com> Date: Mon, 7 Oct 2024 20:26:04 +0200 Subject: [PATCH] `ultralytics 8.3.7` custom Predictor args fix (#16734) Co-authored-by: Glenn Jocher --- ultralytics/__init__.py | 2 +- ultralytics/engine/model.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index ce089ca6b..7ecb2c05a 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.3.6" +__version__ = "8.3.7" import os diff --git a/ultralytics/engine/model.py b/ultralytics/engine/model.py index c4db53426..43b7ccfae 100644 --- a/ultralytics/engine/model.py +++ b/ultralytics/engine/model.py @@ -544,6 +544,8 @@ class Model(nn.Module): if not self.predictor: self.predictor = predictor or self._smart_load("predictor")(overrides=args, _callbacks=self.callbacks) + if predictor: + self.predictor.args = get_cfg(self.predictor.args, args) self.predictor.setup_model(model=self.model, verbose=is_cli) else: # only update args if predictor is already setup self.predictor.args = get_cfg(self.predictor.args, args)