Skip to content
No description, website, or topics provided.
Python PureBasic
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitattributes
README.md
resnet50_inference.py
resnet50_int8.pb

README.md

dawnbench_inference

Prepare Environment

Host: 1 x Nvidia GeForce GTX 1080Ti, 2 x Intel(R) Xeon(R) Silver 4114 CPU @ 2.20GHz, 256G Memory
Container Image: nvcr.io/nvidia/tensorrt:19.05-py3

  1. Pull Docker Image nvcr.io/nvidia/tensorrt:19.05-py3 from NGC
  2. Clone this repo

Run inference

Start container of nvcr.io/nvidia/tensorrt:19.05-py3 and run:

taskset -c 0,2,4,6,8,10,12,14,16,18 python3 resnet50_inference.py \
                                            --model resnet50_int8.pb \
                                            --data_dir /data/dataset/imagenet_tfrecord/validation/ \
                                            --batch_size 1 \
                                            --intra_op_parallelism_threads 1 \
                                            --inter_op_parallelism_threads 10
You can’t perform that action at this time.