HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.3.7 custom Predictor args fix (#16734)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Rick-v-E 2024-10-07 20:26:04 +02:00 • committed by GitHub
parent 2102713540
commit ee5331bdee
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/model.py
View file

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