Skip to content

Commit

Permalink
update README
Browse files Browse the repository at this point in the history
  • Loading branch information
chilo-ms committed Oct 2, 2024
1 parent 06732ed commit ab81c74
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion quantization/image_classification/trt/resnet50/README.md
Original file line number Diff line number Diff line change
@@ -1 +1,6 @@
# ONNX PTQ for yolov3 using TensorRT EP
# ONNX PTQ for using TensorRT EP
Following is the end-to-end example using ORT quantization tool to quantize ONNX model and run/evaluate the quantized model with TRT EP.

## Note
We suggest to use ImageNet 2012 classification dataset to do the model calibration and evaluation. In addition to the sample code we provide below, TensorRT model optimizer which leverages torchvision.datasets already provides
the ability to work with ImageNet dataset.

0 comments on commit ab81c74

Please sign in to comment.