Skip to content

Commit

Permalink
Remove MNIST data. Add instructions how to fetch that
Browse files Browse the repository at this point in the history
  • Loading branch information
tfeher committed Sep 16, 2021
1 parent b41c508 commit 45680b7
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions tf_trt_cpp_example/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,14 @@ python mnist_train.py
```

### Run TF-TRT conversion and infer the converted model
Get input data
```
cd /data/tensorflow-source/tf_trt_cpp_example
wget -O - http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz | gunzip > t10k-images.idx3-ubyte
```
Run inference
```
cd /data/tensorflow-source/tf_trt_cpp_example/build
TF_CPP_VMODULE=trt_convert=2,trt_optimization_pass=2,trt_engine_utils=2,trt_engine_op=2,segment=2,trt_shape_optimization_profiles=2,trt_lru_cache=2,convert_graph=2,trt_engine_resource_ops=2 ./tf_trt_example
```
Binary file removed tf_trt_cpp_example/t10k-images.idx3-ubyte
Binary file not shown.

0 comments on commit 45680b7

Please sign in to comment.