Update utils.py strip_optimizer() (#509)
Follow-on update that I missed adding into PR 509.
This commit is contained in:
1 parent
0032af2980
commit
fe094333e8
1 file changed
+1
-1
@@ -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()
|
||||
Reference in new issue
Block a user