update onnx_export.py usage

2020-06-19 10:15:08 -07:00 · 2020-06-19 10:15:08 -07:00 · e249792c2f
parent 80b82e8bde
commit e249792c2f
1 changed files with 0 additions and 1 deletions
--- a/models/onnx_export.py
+++ b/models/onnx_export.py
@ -1,7 +1,6 @@
 """Exports a pytorch *.pt model to *.onnx format

 Usage:
-    import torch
    $ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
 """