Skip to content

Commit 674a5df

Browse files
authored
Update README.md
1 parent e9ffdef commit 674a5df

File tree

1 file changed

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

1 file changed

+6
-3
lines changed

quantization/image_classification/trt/resnet50/README.md

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
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.
1+
# ONNX PTQ overview
2+
Following is the end-to-end example using ORT quantization tool to quantize ONNX model, specifially image classification model, and run/evaluate the quantized model with TRT EP.
33

44
## Environment setup
55
### dataset
6+
First, prepare dataset for calibration as well as evaluation.
67
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
78
the ability to work with ImageNet dataset.
89

@@ -16,7 +17,7 @@ wget https://image-net.org/data/ILSVRC/2012/ILSVRC2012_devkit_t12.tar.gz --no-ch
1617
```
1718
Untar the tarballs to `val` and `ILSVRC2012_devkit_t12` folder separately.
1819

19-
The dataset layout should look like this: Following sample code expects this dataset layout.
20+
The dataset layout should look like below and the sample code expects this dataset layout
2021

2122
```
2223
|-- ILSVRC2012_devkit_t12
@@ -79,3 +80,5 @@ wget -qO- https://raw.githubusercontent.com/soumith/imagenetloader.torch/master/
7980
...
8081
```
8182

83+
## Quantize an ONNX model
84+

0 commit comments

Comments
 (0)