Skip to content

alipay-cognition/DAWNBench_Inference

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 

DAWNBench_Inference

Resnet101 for DAWNBench inference task on ImageNet

We run Resnet101 on Alibaba Cloud Npu, which consists of 1 NPU.

The following instructions show how to achieve the performance that we submitted to DAWNBench step by step.

  1. Get a Npu container from https://promotion.aliyun.com/ntms/act/alinpu.html

  2. install python3

    wget https://www.python.org/ftp/python/3.5.2/Python-3.5.2.tgz 
    tar -zxf Python-3.5.2.tgz && cd Python-3.5.2 && ./configure && make && sudo make install
  1. install gcc && pillow && numpy && python-gflags && tensorflow
    yum -y install gcc+ gcc-c++
    pip3 install pillow numpy python-gflags tensorflow==1.12
    pip3 install hgai (in npu container from https://promotion.aliyun.com/ntms/act/alinpu.html)
  1. git clone DAWNBench_Inference code
   http://gitlab.alipay-inc.com/CognitiveComputing/dawnbench-cognition
  1. run the following commands to replicate our results submitted to DAWNBench,
   cd inference
   tar xzvf model.tar.gz
   ## 1.download tf records of ImageNet to imagenet_val path
   ## 2.run eval 
   sh eval.sh

5.the result is as follows:

top1:0.750200
top5:0.933800
Latency mean:0.387978ms

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published