Update utils.py strip_optimizer() (#509)

Follow-on update that I missed adding into PR 509.
5 years ago · fe094333e8
parent 0032af2980
commit fe094333e8
1 changed files with 1 additions and 1 deletions
--- 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()