From fe094333e8de3c32313a5447f4f26ea2d0484493 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 24 Jul 2020 19:30:37 -0700
Subject: [PATCH] Update utils.py strip_optimizer() (#509)

Follow-on update that I missed adding into PR 509.
---
 detect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detect.py b/detect.py
index bfe4604..6f4aead 100644
--- a/detect.py
+++ b/detect.py
@@ -156,6 +156,6 @@ if __name__ == '__main__':
         if opt.update:  # update all models (to fix SourceChangeWarning)
             for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
                 detect()
-                create_pretrained(opt.weights, opt.weights)
+                strip_optimizer(opt.weights)
         else:
             detect()