Skip to content

Commit

Permalink
feat: update test speed readme
Browse files Browse the repository at this point in the history
  • Loading branch information
khwengXU committed Jul 5, 2022
1 parent b6cef87 commit 77d1100
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/Test_speed.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,5 +37,5 @@ python deploy/ONNX/export_onnx.py --weights yolov6n.pt --device 0 --batch [1 or
Second, generate an inference trt engine and test speed using `trtexec`:

```
trtexec --onnx=yolov6n.onnx --workspace=1024 --avgRuns=1000 --inputIOFormats=fp16:chw --outputIOFormats=fp16:chw
trtexec --onnx=yolov6n.onnx --workspace=1024 --avgRuns=1000 --inputIOFormats=fp16:chw --outputIOFormats=fp16:chw [--fp16]
```

0 comments on commit 77d1100

Please sign in to comment.