diff --git a/deploy/ONNX/YOLOv6-Dynamic-Batch-onnxruntime.ipynb b/deploy/ONNX/YOLOv6-Dynamic-Batch-onnxruntime.ipynb index d691e155..49c9f604 100644 --- a/deploy/ONNX/YOLOv6-Dynamic-Batch-onnxruntime.ipynb +++ b/deploy/ONNX/YOLOv6-Dynamic-Batch-onnxruntime.ipynb @@ -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 \\" ] }, { @@ -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": { @@ -582,7 +581,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.13" + "version": "3.8.12" } }, "nbformat": 4,