Update utils.py strip_optimizer() (#509)

Follow-on update that I missed adding into PR 509.
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-07-24 19:30:37 -07:00
1 parent 0032af2980
commit fe094333e8
1 file changed
+1 -1
+1 -1
View File
@@ -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()