search

ctuning / ck-mlperf

This repository is outdated! Join the open MLPerf workgroup to participate in the development of the next generation of automation workflows for MLPerf benchmarks:
https://bit.ly/mlperf-edu-wg
BSD 3-Clause "New" or "Revised" License
32 stars 23 forks source link

Support quantized MobileNet-v2 models #46

Closed psyhtest closed 4 years ago

psyhtest commented 4 years ago

For v0.5, we only used quantized MobileNet-v1 models, but not to quantized MobileNet-v2 ones. Google have since released quantized MobileNet-v2 checkpoints. We should investigate if they can be converted to quantized TFLite models relatively easily (if they are not already).

psyhtest commented 4 years ago

MobileNet-v2-1.3-224 is not available but otherwise they seem to work fine now with a new package. The new package is separate from the non-quantized one due to unexpected and inconvenient changes to the filenames. Also, the output layer name had to be updated:

      "MODEL_OUTPUT_LAYER_NAME": "MobilenetV2/Predictions/Softmax",