Skip to content

Commit

Permalink
Move limitations to the top
Browse files Browse the repository at this point in the history
  • Loading branch information
BloodAxe committed May 28, 2024
1 parent 0edba4d commit d3d4be1
Showing 1 changed file with 12 additions and 21 deletions.
33 changes: 12 additions & 21 deletions notebooks/YoloNAS_R_Export_to_ONNX.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -48,9 +48,20 @@
"- Customising input image shape and batch size\n",
"- Customising NMS parameters and number of detections per image\n",
"- Customising output format (flat or batched)\n",
"- Exported ONNX model can be used with TensorRT"
"- Exported ONNX model can be used with TensorRT\n",
"\n",
"### Limitations\n",
"\n",
"- Dynamic batch size / input image shape is not supported yet. You can only export a model with a fixed batch size and input image shape.\n",
"- TensorRT of version 8.6 or higher is required.\n",
"- Quantization to FP16 requires CUDA / MPS device available."
]
},
{
"metadata": {},
"cell_type": "markdown",
"source": ""
},
{
"cell_type": "code",
"execution_count": 1,
Expand Down Expand Up @@ -1125,26 +1136,6 @@
"show_predictions_from_flat_format(image, result)"
]
},
{
"cell_type": "markdown",
"metadata": {
"collapsed": false,
"id": "KZK3oB3EG7ba",
"jupyter": {
"outputs_hidden": false
},
"pycharm": {
"name": "#%% md\n"
}
},
"source": [
"### Limitations\n",
"\n",
"* Dynamic batch size / input image shape is not supported yet. You can only export a model with a fixed batch size and input image shape.\n",
"* TensorRT of version 8.6 or higher is required.\n",
"* Quantization to FP16 requires CUDA / MPS device available."
]
},
{
"cell_type": "markdown",
"metadata": {
Expand Down

0 comments on commit d3d4be1

Please sign in to comment.