glenn-jocher commited on
Commit
fe09433
1 Parent(s): 0032af2

Update utils.py strip_optimizer() (#509)

Browse files

Follow-on update that I missed adding into PR 509.

Files changed (1) hide show
  1. detect.py +1 -1
detect.py CHANGED
@@ -156,6 +156,6 @@ if __name__ == '__main__':
156
  if opt.update: # update all models (to fix SourceChangeWarning)
157
  for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
158
  detect()
159
- create_pretrained(opt.weights, opt.weights)
160
  else:
161
  detect()
 
156
  if opt.update: # update all models (to fix SourceChangeWarning)
157
  for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
158
  detect()
159
+ strip_optimizer(opt.weights)
160
  else:
161
  detect()