glenn-jocher commited on
Commit
36f64a9
1 Parent(s): 1ab23fc

Update tutorial.ipynb (#8507)

Browse files
Files changed (1) hide show
  1. tutorial.ipynb +2 -2
tutorial.ipynb CHANGED
@@ -1105,8 +1105,8 @@
1105
  "# TensorRT \n",
1106
  "# https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-pip\n",
1107
  "!pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com # install\n",
1108
- "!python export.py --weights yolov5s.pt --include engine --imgsz 640 640 --device 0 # export\n",
1109
- "!python detect.py --weights yolov5s.engine --imgsz 640 640 --device 0 # inference"
1110
  ],
1111
  "execution_count": null,
1112
  "outputs": []
 
1105
  "# TensorRT \n",
1106
  "# https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-pip\n",
1107
  "!pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com # install\n",
1108
+ "!python export.py --weights yolov5s.pt --include engine --imgsz 640 --device 0 # export\n",
1109
+ "!python detect.py --weights yolov5s.engine --imgsz 640 --device 0 # inference"
1110
  ],
1111
  "execution_count": null,
1112
  "outputs": []