Skip to content

Commit

Permalink
update the command of exporting model to onnx.
Browse files Browse the repository at this point in the history
  • Loading branch information
mtjhl committed Nov 16, 2022
1 parent e302cc8 commit c59ef99
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions deploy/ONNX/YOLOv6-Dynamic-Batch-onnxruntime.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -77,8 +77,7 @@
" --iou-thres 0.65 \\\n",
" --conf-thres 0.35 \\\n",
" --img-size 640 640 \\\n",
" --dynamic-batch \\\n",
" --max-wh 7680"
" --dynamic-batch \\"
]
},
{
Expand Down Expand Up @@ -568,9 +567,9 @@
],
"metadata": {
"kernelspec": {
"display_name": "Python [conda env:torch] *",
"display_name": "Python 3 (ipykernel)",
"language": "python",
"name": "conda-env-torch-py"
"name": "python3"
},
"language_info": {
"codemirror_mode": {
Expand All @@ -582,7 +581,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.13"
"version": "3.8.12"
}
},
"nbformat": 4,
Expand Down

0 comments on commit c59ef99

Please sign in to comment.