Skip to content

Commit d3eab45

Browse files
authored
Update README.md
1 parent c628f0c commit d3eab45

File tree

1 file changed

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

1 file changed

+7
-1
lines changed

quantization/image_classification/trt/resnet50/README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,4 +85,10 @@ wget -qO- https://raw.githubusercontent.com/soumith/imagenetloader.torch/master/
8585
Lastly, download `synset_words.txt` from https://github.com/HoldenCaulfieldRye/caffe/blob/master/data/ilsvrc12/synset_words.txt into `ILSVRC2012` (top-level folder)
8686

8787
## Quantize an ONNX model
88-
88+
```shell
89+
python e2e_tensorrt_resnet_example.py
90+
````
91+
The e2e script will do
92+
* Preprocess the image in the dataset
93+
* Run calibration and generate calibration table (deprecated)
94+
* Launch ORT session with TRT EP and run the evaluation (top 1 and top 5)

0 commit comments

Comments
 (0)