pretty code

2020年6月29日 星期一

SSD-ResNet34 model on MLPerf v0.5

In the official web site, the format of the link of SSD-ResNet34 model is "NCHW". This format is usually for GPU. It means that we can not use it on TensorFlow.

I found a link after searching google. It seems that the format is "NHWC". I used this mode to run MLPerf v0.5. I got the result finally.

The link is shown below.
https://zenodo.org/record/3246481#.XvmdbCgzaUk


I used these below parameters to run MLPerf.
./run_local.sh tf ssd-resnet34 cpu --data-format NHWC

Although I got the result, there was errors of "only size-1 arrays can be converted to Python scalars". I will check it later.


Reference

https://github.com/mlperf/inference/issues/219
https://github.com/mlperf/inference/issues/229

沒有留言: