Unverified 提交 fe094333 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update utils.py strip_optimizer() (#509)

Follow-on update that I missed adding into PR 509.
上级 0032af29
...@@ -156,6 +156,6 @@ if __name__ == '__main__': ...@@ -156,6 +156,6 @@ if __name__ == '__main__':
if opt.update: # update all models (to fix SourceChangeWarning) if opt.update: # update all models (to fix SourceChangeWarning)
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']: for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
detect() detect()
create_pretrained(opt.weights, opt.weights) strip_optimizer(opt.weights)
else: else:
detect() detect()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论