Fix detect.py --update list bug (ultralytics#8783)

Fix detect.py --update Resolves ultralytics#8776
ctjanuhowski · Sep 8, 2022 · 6fd9ea1 · 6fd9ea1
1 parent 0537991
commit 6fd9ea1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detect.py b/detect.py
@@ -210,7 +210,7 @@ def run(
         s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
         LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}{s}")
     if update:
-        strip_optimizer(weights)  # update model (to fix SourceChangeWarning)
+        strip_optimizer(weights[0])  # update model (to fix SourceChangeWarning)
 
 
 def parse_opt():