Skip to content

Commit ab81c74

Browse files
committed
update README
1 parent 06732ed commit ab81c74

File tree

1 file changed

+6
-1
lines changed
  • quantization/image_classification/trt/resnet50

1 file changed

+6
-1
lines changed
Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,6 @@
1-
# ONNX PTQ for yolov3 using TensorRT EP
1+
# ONNX PTQ for using TensorRT EP
2+
Following is the end-to-end example using ORT quantization tool to quantize ONNX model and run/evaluate the quantized model with TRT EP.
3+
4+
## Note
5+
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
6+
the ability to work with ImageNet dataset.

0 commit comments

Comments
 (0)