AberHu / TF-NAS

TF-NAS: Rethinking Three Search Freedoms of Latency-Constrained Differentiable Neural Architecture Search (ECCV2020)
https://arxiv.org/abs/2008.05314
MIT License
76 stars 11 forks source link

Could you open source code of making latency lookup tables? #2

Open touchdreamer opened 4 years ago

AberHu commented 4 years ago

Hi, please refer to make_lat_lut_example.py. It is an example script to make latency lookup tables. Due to the fine-grained width search, we enumerate all possible width choices, which is time consuming especially for CPUs. You can also enumerate width choices every 8 steps and make interpolation for other choices as a rough approximation, referring to the comments in the script.

Note, before running the script, please set the number of threads for OpenMP and MKL to 1 and 1, respectively.

touchdreamer commented 4 years ago

Thanks a lot.