glenn-jocher
commited on
Commit
•
fe09433
1
Parent(s):
0032af2
Update utils.py strip_optimizer() (#509)
Browse filesFollow-on update that I missed adding into PR 509.
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 |
-
|
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()
|