`ultralytics 8.3.7` custom Predictor args fix (#16734)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/16748/head v8.3.7
Rick-v-E 4 months ago committed by GitHub
parent 2102713540
commit ee5331bdee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      ultralytics/__init__.py
  2. 2
      ultralytics/engine/model.py

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = "8.3.6"
__version__ = "8.3.7"
import os

@ -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)

Loading…
Cancel
Save