Skip to content

Commit

Permalink
update onnx_export.py usage
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Jun 19, 2020
1 parent 80b82e8 commit e249792
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion models/onnx_export.py
Original file line number Diff line number Diff line change
@@ -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
"""

Expand Down

0 comments on commit e249792

Please sign in to comment.